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 ด้วย)

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

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

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

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

RaggedGather <T, U> แบบคงที่สาธารณะ สร้าง (ขอบเขต ขอบเขต , Iterable< Operand <T>> paramsNestedSplits, Operand <U> paramsDenseValues, Operand <V> ดัชนี, 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 ที่ส่งคืน