RaggedGather

kelas akhir publik RaggedGather

Kumpulkan irisan kasar dari sumbu `params` `0` menurut `indeks`.

Menghasilkan keluaran `RaggedTensor` yang terdiri dari `output_dense_values` dan `output_nested_splits`, sehingga:

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]
 
di mana
  • `params = ragged.from_nested_row_splits(params_dense_values, params_nested_splits)` memberikan nilai yang harus dikumpulkan.
  • `indices` merupakan tensor padat dengan dtype `int32` atau `int64`, yang menunjukkan nilai mana yang harus dikumpulkan.
  • `output = ragged.from_nested_row_splits(output_dense_values, output_nested_splits)` adalah tensor keluaran.
(Catatan: Operasi c++ ini digunakan untuk mengimplementasikan operasi python `tf.ragged.gather` tingkat tinggi, yang juga mendukung indeks ragged.)

Metode Publik

statis <T memperluas Nomor, U, V memperluas Nomor> RaggedGather <T, U>
buat ( Lingkup lingkup, Iterable< Operand <T>> paramsNestedSplits, Operand <U> paramsDenseValues, Operand <V> indeks, Long OUTPUTRAGGEDRANK)
Metode pabrik untuk membuat kelas yang membungkus operasi RaggedGather baru.
Keluaran <U>
keluaranDenseValues ()
`flat_values` untuk RaggedTensor yang dikembalikan.
Daftar< Keluaran <T>>
keluaranNestedSplits ()
Tensor `nested_row_splits` yang menentukan partisi baris untuk RaggedTensor yang dikembalikan.

Metode Warisan

Metode Publik

public static RaggedGather <T, U> buat ( Scope scope, Iterable< Operand <T>> paramsNestedSplits, Operand <U> paramsDenseValues, Operand <V> indeks, Long OUTPUTRAGGEDRANK)

Metode pabrik untuk membuat kelas yang membungkus operasi RaggedGather baru.

Parameter
cakupan ruang lingkup saat ini
paramsNestedSplits Tensor `nested_row_splits` yang menentukan partisi baris untuk input `params` RaggedTensor.
paramsDenseValues `flat_values` untuk `params` RaggedTensor. Ada perubahan terminologi di level python dari padat_nilai menjadi nilai_datar, jadi nilai_padat adalah nama yang tidak digunakan lagi.
indeks Indeks di dimensi terluar `params` dari nilai yang harus dikumpulkan.
OUTPUTRAGGEDRANK Peringkat keluaran RaggedTensor yang tidak rata. `output_nested_splits` akan berisi jumlah tensor `row_splits` ini. Nilai ini harus sama dengan `indices.shape.ndims + params.ragged_rank - 1`.
Kembali
  • contoh baru dari RaggedGather

Keluaran publik <U> keluaranDenseValues ​​()

`flat_values` untuk RaggedTensor yang dikembalikan.

Daftar publik< Keluaran <T>> keluaranNestedSplits ()

Tensor `nested_row_splits` yang menentukan partisi baris untuk RaggedTensor yang dikembalikan.