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.
Metode Publik
static <T perpanjang Angka, U, V perpanjang Angka> RaggedGather <T, U> | |
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
Daftar publik< Output <T>> outputNestedSplits ()
Tensor `nested_row_splits` yang menentukan partisi baris untuk RaggedTensor yang dikembalikan.