パブリック最終クラスRaggedGather
`params` 軸 `0` から `indices` に従って不規則なスライスを収集します。
次のような `output_dense_values` と `output_nested_splits` から構成される `RaggedTensor` 出力を出力します。
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` は、dtype `int32` または `int64` を持つ密なテンソルであり、どの値を収集する必要があるかを示します。
- `output = ragged.from_nested_row_splits(output_dense_values, Output_nested_splits)` は出力テンソルです。
パブリックメソッド
static <T extends Number, U, V extends Number> RaggedGather <T, U> | |
出力<U> | 出力DenseValues () 返された RaggedTensor の ` flat_values `。 |
リスト<出力<T>> | 出力NestedSplits () 返された RaggedTensor の行分割を定義する `nested_row_splits` テンソル。 |
継承されたメソッド
パブリックメソッド
public static RaggedGather <T, U> create (スコープスコープ、Iterable<オペランド<T>> paramsNestedSplits、オペランド<U> paramsDenseValues、オペランド<V> インデックス、Long OUTPUTRAGGEDRANK)
新しい RaggedGather オペレーションをラップするクラスを作成するためのファクトリ メソッド。
パラメータ
範囲 | 現在のスコープ |
---|---|
paramsネストされた分割 | `params` RaggedTensor 入力の行分割を定義する `nested_row_splits` テンソル。 |
paramsDenseValues | `params` RaggedTensor の ` flat_values `。 Python レベルで用語がdense_values から flat_values に変更されたため、dense_values は非推奨の名前です。 |
インデックス | 収集する必要がある値の「params」の最も外側の次元のインデックス。 |
OUTPUTTRAGGEDRANK | 出力 RaggedTensor の不規則なランク。 `output_nested_splits` には、この数の `row_splits` テンソルが含まれます。この値は、「indices.shape.ndims + params.ragged_rank - 1」と等しくなければなりません。 |
返品
- RaggedGather の新しいインスタンス