パブリック最終クラスXlaRecvTPUEmbeddingDeduplicationData
組み込みコアから重複排除データ (インデックスと重み) を受信します。
重複排除データは、type=DT_VARIANT の Tensor です。テンソル自体は、N 個の要素を含む XLA ネストされたタプルです (N は、TPU チップあたりのテンソル コアに対する埋め込みの数の比率です)。入れ子になったタプルの各要素は、ランク 1 テンソルのタプルです。各テンソルには、TensorCore での埋め込みルックアップ用のインデックス (DT_UINT32) または埋め込みルックアップ操作の出力に適用する重み (DT_FLOAT) が含まれています。
パブリックメソッド
| 出力<オブジェクト> | asOutput () テンソルのシンボリック ハンドルを返します。 |
| 静的XlaRecvTPUEmbeddingDeduplicationData | |
| 出力<?> | 出力() |
継承されたメソッド
パブリックメソッド
public Output <オブジェクト> asOutput ()
テンソルのシンボリック ハンドルを返します。
TensorFlow オペレーションへの入力は、別の TensorFlow オペレーションの出力です。このメソッドは、入力の計算を表すシンボリック ハンドルを取得するために使用されます。
public static XlaRecvTPUEmbeddingDeduplicationData create (スコープスコープ、文字列構成)
新しい XlaRecvTPUEmbeddingDeduplicationData 操作をラップするクラスを作成するファクトリ メソッド。
パラメーター
| 範囲 | 現在のスコープ |
|---|---|
| 構成 | シリアル化された TPUEmbeddingConfiguration プロト。 |
戻り値
- XlaRecvTPUEmbeddingDeduplicationData の新しいインスタンス