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のオプション属性

パブリックメソッド

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

継承されたメソッド

パブリックメソッド

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

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

public static EnqueueTPUEmbeddingSparseBatch create (スコープスコープ、Iterable<オペランド<T>> サンプルインデックス、Iterable<オペランド<U>> embeddingIndices、Iterable<オペランド<V>> aggregationWeights、オペランド<String> 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 コアの数未満である必要があります。