パブリック最終クラスEnqueueTPUEmbeddingBatch
入力バッチ テンソルのリストを TPUEmbedding にエンキューする操作。
入力バッチ テンソルのリストを TPUEmbedding にエンキューする操作。
ネストされたクラス
| クラス | EnqueueTPUEmbeddingBatch.Options | EnqueueTPUEmbeddingBatchのオプションの属性 | |
パブリックメソッド
| 静的EnqueueTPUEmbeddingBatch.Options | コンバイナー(List<String> コンバイナー) |
| 静的EnqueueTPUEmbeddingBatch | |
| 静的EnqueueTPUEmbeddingBatch.Options | deviceOrdinal (長い deviceOrdinal) |
継承されたメソッド
パブリックメソッド
public static EnqueueTPUEmbeddingBatch.Optionsコンバイナー(List<String> コンバイナー)
パラメーター
| 結合器 | 文字列スカラーのリスト。重み付き合計後の埋め込みアクティブ化を正規化する方法を指定する各埋め込みテーブルに 1 つあります。サポートされている結合子は、「mean」、「sum」、または「sqrtn」です。 'mean' の重みの合計が 0 になること、または 'sqrtn' の 2 乗重みの合計が 0 になることは無効です。コンバイナーが渡されない場合、デフォルトではすべてのテーブルに「sum」が使用されます。 |
|---|
public static EnqueueTPUEmbeddingBatch create (スコープスコープ、Iterable<オペランド<String>> バッチ、オペランド<String> modeOverride、オプション...オプション)
新しい EnqueueTPUEmbeddingBatch オペレーションをラップするクラスを作成するファクトリ メソッド。
パラメーター
| 範囲 | 現在のスコープ |
|---|---|
| バッチ | 1D テンソルのリスト (埋め込みテーブルごとに 1 つ)。dist_belief.SparseFeature プロトとしてエンコードされたバッチ入力が含まれます。 SparseFeature プロトの重みフィールドに ID が設定されていない場合は、重み 1.0 が想定されます。 |
| モードオーバーライド | TPUEmbeddingConfiguration で指定されたモードをオーバーライドする文字列入力。サポートされている値は、{'unspecified'、'inference'、'training'、'backward_pass_only'} です。 「unspecified」に設定すると、TPUEmbeddingConfiguration で設定されたモードが使用され、それ以外の場合は、mode_override が使用されます。 |
| オプション | オプションの属性値を持ちます |
戻り値
- EnqueueTPUEmbeddingBatch の新しいインスタンス
public static EnqueueTPUEmbeddingBatch.Options deviceOrdinal (Long deviceOrdinal)
パラメーター
| デバイス通常 | 使用する TPU デバイス。これは、Op が TPU デバイスで実行されている場合は -1 である必要があり、Op が CPU デバイスで実行されている場合は >= 0 である必要があります。 |
|---|