パブリック最終クラスDynamicEnqueueTPUEmbeddingArbitraryTensorBatch
tf.nn.embedding_lookup_sparse() を使用するコードの移植を容易にします。
embedding_indices[i] と aggregation_weights[i] が i 番目の特徴に対応します。
3 つの入力リスト (sample_indices、embedding_indices、および aggregation_weights) 内の対応する位置にあるテンソルは同じ形状、つまり対応する特徴によって記述されるテーブルへのルックアップの総数に等しい dim_size() を持つランク 1 を持つ必要があります。
ネストされたクラス
| クラス | DynamicEnqueueTPUEmbeddingArbitraryTensorBatch.Options | DynamicEnqueueTPUEmbeddingArbitraryTensorBatchのオプション属性 | |
パブリックメソッド
| 静的DynamicEnqueueTPUEmbeddingArbitraryTensorBatch.Options | コンバイナー(List<String> コンバイナー) |
| static <T extends Number, U extends Number, V extends Number> DynamicEnqueueTPUEmbeddingArbitraryTensorBatch | create ( Scopeスコープ、Iterable< Operand <T>> sampleIndicesOrRowSplits、Iterable< Operand <U>> embeddingIndices、Iterable< Operand <V>> aggregationWeights、 Operand <String> modeOverride、 Operand <Integer> deviceOrdinal、 Options... options) 新しい DynamicEnqueueTPUEmbeddingArbitraryTensorBatch オペレーションをラップするクラスを作成するファクトリ メソッド。 |
継承されたメソッド
パブリックメソッド
public static DynamicEnqueueTPUEmbeddingArbitraryTensorBatch.Optionsコンバイナー(List<String> コンバイナー)
パラメーター
| 結合器 | 文字列スカラーのリスト。重み付き合計後の埋め込みアクティブ化を正規化する方法を指定する各埋め込みテーブルに 1 つあります。サポートされている結合子は、「mean」、「sum」、または「sqrtn」です。 'mean' の重みの合計が 0 になること、または 'sqrtn' の 2 乗重みの合計が 0 になることは無効です。コンバイナーが渡されない場合、デフォルトではすべてのテーブルに「sum」が使用されます。 |
|---|
public static DynamicEnqueueTPUEmbeddingArbitraryTensorBatch create ( Scopeスコープ、Iterable< Operand <T>> sampleIndicesOrRowSplits、Iterable< Operand <U>> embeddingIndices、Iterable< Operand <V>> aggregationWeights、 Operand <String> modeOverride、 Operand <Integer> deviceOrdinal、 Options.. .オプション)
新しい DynamicEnqueueTPUEmbeddingArbitraryTensorBatch オペレーションをラップするクラスを作成するファクトリ メソッド。
パラメーター
| 範囲 | 現在のスコープ |
|---|---|
| サンプルインデックスまたは行分割 | 対応する embedding_indices および aggregation_weights 値が属するトレーニング サンプルを指定するランク 2 テンソルのリスト。最初の次元のサイズが 0 の場合、各 embedding_indices は異なるサンプルに属していると想定されます。 int32 と int64 の両方が許可され、内部的に int32 に変換されます。 または、embedding_indices と aggregation_weights を行に分割するための行分割を指定するランク 1 Tensor のリスト。 ids が RaggedTensor の場合、embedding_lookup() の ids.row_splits に対応します。 ND 不規則テンソルをキューに入れる場合、最後の次元のみ不規則化が許可されます。行分割は 1 次元の密なテンソルです。空の場合、密なテンソルが op に渡されると想定します。 int32 と int64 の両方が許可され、内部で int32 に変換されます。 |
| 埋め込みインデックス | ランク 1 の Tensor、埋め込みテーブルへのインデックスのリスト。 int32 と int64 の両方が許可され、内部的に int32 に変換されます。 |
| 集計重み | トレーニング サンプルごとの集約重みを含むランク 1 テンソルのリスト。 float32 と float64 の両方が許可されており、内部的に float32 に変換されます。 |
| モードオーバーライド | TPUEmbeddingConfiguration で指定されたモードをオーバーライドする文字列入力。サポートされている値は、{'unspecified'、'inference'、'training'、'backward_pass_only'} です。 「unspecified」に設定すると、TPUEmbeddingConfiguration で設定されたモードが使用され、それ以外の場合は、mode_override が使用されます。 |
| デバイス通常 | 使用する TPU デバイス。 0 以上で、ノードが配置されているタスク内の TPU コアの数未満である必要があります。 |
| オプション | オプションの属性値を持ちます |
戻り値
- DynamicEnqueueTPUEmbeddingArbitraryTensorBatch の新しいインスタンス