パブリック最終クラス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 | |
静的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 コアの数未満である必要があります。 |
---|