不規則に集まる

パブリック最終クラス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)` は出力テンソルです。
(注: この C++ 演算は、不規則なインデックスもサポートする高レベルの Python `tf.ragged.gather` 演算を実装するために使用されます。)

パブリックメソッド

static <T extends Number, U, V extends Number> RaggedGather <T, U>
create (スコープスコープ、Iterable<オペランド<T>> paramsNestedSplits、オペランド<U> paramsDenseValues、オペランド<V> インデックス、Long OUTPUTRAGGEDRANK)
新しい RaggedGather オペレーションをラップするクラスを作成するためのファクトリ メソッド。
出力<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 の新しいインスタンス

public出力<U> OutputDenseValues ()

返された RaggedTensor の ` flat_values `。

public List<出力<T>> OutputNestedSplits ()

返された RaggedTensor の行分割を定義する `nested_row_splits` テンソル。