Peringatan: API ini tidak digunakan lagi dan akan dihapus di versi TensorFlow mendatang setelah penggantinya stabil.

RaggedGather

kelas akhir publik RaggedGather

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

Mengeluarkan 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 output.
(Catatan: Operasi c++ ini digunakan untuk mengimplementasikan operasi python `tf.ragged.gather` tingkat tinggi, yang juga mendukung indeks kasar.)

Metode Publik

static <T perpanjang Angka, U, V perpanjang Angka> RaggedGather <T, U>
buat ( Cakupan cakupan, Iterable< Operand <T>> paramsNestedSplits, Operand <U> paramsDenseValues, indeks Operand <V>, OUTPUTRAGGEDRANK Panjang)
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 yang Diwarisi

Metode Publik

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

Metode pabrik untuk membuat kelas yang membungkus operasi RaggedGather baru.

Parameter
cakupan lingkup saat ini
paramsNestedSplits Tensor `nested_row_splits` yang menentukan partisi baris untuk input RaggedTensor `params`.
paramsDenseValues `flat_values` untuk `params` RaggedTensor. Ada perubahan terminologi di tingkat python dari density_values ​​ke flat_values, jadi density_values ​​adalah nama yang tidak digunakan lagi.
indeks Indeks dalam dimensi terluar `params` dari nilai yang harus dikumpulkan.
OUTPUTRAGGEDRANK Peringkat kasar dari output RaggedTensor. `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

Output publik <U> outputDenseValues ()

`flat_values` untuk RaggedTensor yang dikembalikan.

Daftar publik< Output <T>> outputNestedSplits ()

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