EnqueueTPUEmbeddingSparseBatch

パブリック最終クラスEnqueueTPUEmbeddingSparseBatch

SparseTensor からの TPUEmbedding 入力インデックスをキューに入れる操作。

この Op により、embedding_lookup_sparse() を使用するコードの移植が容易になりますが、トレーニング ステップごとに許可される EnqueueTPUEmbeddingSparseBatch Op は 1 つだけであるため、この Op への引数を生成するには、embedding_lookup_sparse() への SparseTensor 引数の Python 前処理が必要になります。

3 つの入力リストの対応する位置にあるテンソルは同じ形状でなければなりません。つまり、対応する table_id によって記述されるテーブルへのルックアップの総数に等しい dim_size() を持つランク 1 でなければなりません。

ネストされたクラス

クラスEnqueueTPUEmbeddingSparseBatch.Options EnqueueTPUEmbeddingSparseBatchのオプション属性

定数

OP_NAME TensorFlow コア エンジンによって認識される、この演算の名前

パブリックメソッド

静的EnqueueTPUEmbeddingSparseBatch.Options
コンバイナー(List<String> コンバイナー)
static <T extends TNumber 、U extends TNumber 、V extends TNumber > EnqueueTPUEmbeddingSparseBatch
create ( Scopeスコープ、Iterable< Operand <T>> sampleIndices、Iterable< Operand <U>> embeddingIndices、Iterable< Operand <V>> aggregationWeights、 Operand < TString > modeOverride、 Options... options)
新しい EnqueueTPUEmbeddingSparseBatch オペレーションをラップするクラスを作成するファクトリ メソッド。
静的EnqueueTPUEmbeddingSparseBatch.Options
deviceOrdinal (長い deviceOrdinal)

継承されたメソッド

定数

パブリック静的最終文字列OP_NAME

TensorFlow コア エンジンによって認識される、この演算の名前

定数値: "EnqueueTPUEmbeddingSparseBatch"

パブリックメソッド

public static EnqueueTPUEmbeddingSparseBatch.Optionsコンバイナー(List<String> コンバイナー)

パラメーター
結合器文字列スカラーのリスト。重み付き合計後の埋め込みアクティブ化を正規化する方法を指定する各埋め込みテーブルに 1 つあります。サポートされている結合子は、「mean」、「sum」、または「sqrtn」です。 'mean' の重みの合計が 0 になること、または 'sqrtn' の 2 乗重みの合計が 0 になることは無効です。コンバイナーが渡されない場合、デフォルトではすべてのテーブルに「sum」が使用されます。

public static EnqueueTPUEmbeddingSparseBatch create ( Scopeスコープ、Iterable< Operand <T>> sampleIndices、Iterable< Operand <U>> embeddingIndices、Iterable< Operand <V>> aggregationWeights、 Operand < TString > modeOverride、 Options...オプション)

新しい EnqueueTPUEmbeddingSparseBatch オペレーションをラップするクラスを作成するファクトリ メソッド。

パラメーター
範囲現在のスコープ
サンプルインデックス対応するembedding_indicesとaggregation_weightsの値が属するトレーニング例と機能を指定するランク1テンソルのリスト。 Sample_indices[i] は b * nf + f に等しくなければなりません。ここで、nf は対応するテーブルの特徴の数、f は [0, nf) にあり、b は [0, バッチ サイズ) にあります。
埋め込みインデックスランク 1 の Tensor、埋め込みテーブルへのインデックスのリスト。
集計重みサンプルごと、つまり (トレーニング例、機能) ごとの集計重みを含むランク 1 テンソルのリスト。
モードオーバーライドTPUEmbeddingConfiguration で指定されたモードをオーバーライドする文字列入力。サポートされている値は、{'unspecified'、'inference'、'training'、'backward_pass_only'} です。 「unspecified」に設定すると、TPUEmbeddingConfiguration で設定されたモードが使用され、それ以外の場合は、mode_override が使用されます。
オプションオプションの属性値を持ちます
戻り値
  • EnqueueTPUEmbeddingSparseBatch の新しいインスタンス

public static EnqueueTPUEmbeddingSparseBatch.Options deviceOrdinal (Long deviceOrdinal)

パラメーター
デバイス通常使用する TPU デバイス。 0 以上で、ノードが配置されているタスク内の TPU コアの数未満である必要があります。