RaggedGather

RaggedGather คลาสสุดท้ายสาธารณะ

รวบรวมส่วนที่ขาดจากแกน `params` `0` ตาม `ดัชนี`

เอาต์พุตเอาต์พุต `RaggedTensor` ที่ประกอบด้วย `output_dense_values` และ `output_nested_splits` เช่น:

output.shape = indices.shape + params.shape[1:]
 output.ragged_rank = indices.shape.ndims + params.ragged_rank
 output[i...j, d0...dn] = params[indices[i...j], d0...dn]
ที่ไหน
  • `params = ragged.from_nested_row_splits(params_dense_values, params_nested_splits)` ให้ค่าที่ควรรวบรวม
  • `indices` คือเทนเซอร์หนาแน่นที่มีประเภท `int32` หรือ `int64` ซึ่งระบุว่าควรรวบรวมค่าใด
  • `output = ragged.from_nested_row_splits(output_dense_values, output_nested_splits)` คือเทนเซอร์เอาท์พุต
(หมายเหตุ: c++ op นี้ใช้เพื่อใช้งาน python `tf.ragged.gather` op ในระดับที่สูงกว่า ซึ่งรองรับดัชนี ragged ด้วย)

ค่าคงที่

สตริง OP_NAME ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow

วิธีการสาธารณะ

คงที่ <T ขยาย TNumber , U ขยาย TType > RaggedGather <T, U>
สร้าง ( ขอบเขต ขอบเขต Iterable< Operand <T>> paramsNestedSplits, Operand <U> paramsDenseValues, Operand <? ขยาย TNumber > ดัชนี, Long OUTPUTRAGGEDRANK)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ RaggedGather ใหม่
เอาท์พุต <U>
เอาท์พุทDenseValues ​​()
`flat_values` สำหรับ RaggedTensor ที่ส่งคืน
รายการ < เอาท์พุต <T>>
เอาท์พุท NestedSplits ()
เทนเซอร์ `nested_row_splits` ที่กำหนดการแบ่งพาร์ติชันแถวสำหรับ RaggedTensor ที่ส่งคืน

วิธีการสืบทอด

ค่าคงที่

สตริงสุดท้ายแบบคงที่สาธารณะ OP_NAME

ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow

ค่าคงที่: "RaggedGather"

วิธีการสาธารณะ

สาธารณะ RaggedGather <T, U> สร้าง แบบคงที่ (ขอบเขต ขอบเขต , Iterable< Operand <T>> paramsNestedSplits, Operand <U> paramsDenseValues, Operand <? ขยาย TNumber > ดัชนี, Long OUTPUTRAGGEDRANK)

วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ RaggedGather ใหม่

พารามิเตอร์
ขอบเขต ขอบเขตปัจจุบัน
paramsNestedSplits เทนเซอร์ `nested_row_splits` ที่กำหนดการแบ่งพาร์ติชันแถวสำหรับอินพุต `params` RaggedTensor
paramsDenseValues `flat_values` สำหรับ `params` RaggedTensor มีการเปลี่ยนแปลงคำศัพท์ในระดับหลามจากหนาแน่น_ค่าไปเป็นค่าแบนดังนั้นหนาแน่น_ค่าจึงเป็นชื่อที่เลิกใช้แล้ว
ดัชนี ดัชนีในมิติด้านนอกสุดของ "พารามิเตอร์" ของค่าที่ควรรวบรวม
เอาท์พุตดรางค์ อันดับ Ragged ของเอาต์พุต RaggedTensor `output_nested_splits` จะมีเทนเซอร์ `row_splits` จำนวนนี้ ค่านี้ควรเท่ากับ `indices.shape.ndims + params.ragged_rank - 1`
การส่งคืน
  • ตัวอย่างใหม่ของ RaggedGather

เอาท์พุท สาธารณะ <U> outputDenseValues ​​()

`flat_values` สำหรับ RaggedTensor ที่ส่งคืน

รายการสาธารณะ < เอาท์พุต <T>> เอาท์พุต NestedSplits ()

เทนเซอร์ `nested_row_splits` ที่กำหนดการแบ่งพาร์ติชันแถวสำหรับ RaggedTensor ที่ส่งคืน