EnqueueTPUEmbeddingBatch

パブリック最終クラスEnqueueTPUEmbeddingBatch

入力バッチ テンソルのリストを TPUEmbedding にエンキューする操作。

入力バッチ テンソルのリストを TPUEmbedding にエンキューする操作。

ネストされたクラス

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

パブリックメソッド

静的EnqueueTPUEmbeddingBatch.Options
コンバイナー(List<String> コンバイナー)
静的EnqueueTPUEmbeddingBatch
create (スコープスコープ、Iterable<オペランド<String>> バッチ、オペランド<String> modeOverride、オプション...オプション)
新しい 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 である必要があります。