XlaRecvTPUEmbeddingDeduplicationData

パブリック最終クラスXlaRecvTPUEmbeddingDeduplicationData

組み込みコアから重複排除データ (インデックスと重み) を受信します。

重複排除データは、type=DT_VARIANT の Tensor です。テンソル自体は、N 個の要素を含む XLA ネストされたタプルです (N は、TPU チップあたりのテンソル コアに対する埋め込みの数の比率です)。入れ子になったタプルの各要素は、ランク 1 テンソルのタプルです。各テンソルには、TensorCore での埋め込みルックアップ用のインデックス (DT_UINT32) または埋め込みルックアップ操作の出力に適用する重み (DT_FLOAT) が含まれています。

パブリックメソッド

出力<オブジェクト>
asOutput ()
テンソルのシンボリック ハンドルを返します。
静的XlaRecvTPUEmbeddingDeduplicationData
create (スコープスコープ、文字列構成)
新しい XlaRecvTPUEmbeddingDeduplicationData 操作をラップするクラスを作成するファクトリ メソッド。
出力<?>

継承されたメソッド

パブリックメソッド

public Output <オブジェクト> asOutput ()

テンソルのシンボリック ハンドルを返します。

TensorFlow オペレーションへの入力は、別の TensorFlow オペレーションの出力です。このメソッドは、入力の計算を表すシンボリック ハンドルを取得するために使用されます。

public static XlaRecvTPUEmbeddingDeduplicationData create (スコープスコープ、文字列構成)

新しい XlaRecvTPUEmbeddingDeduplicationData 操作をラップするクラスを作成するファクトリ メソッド。

パラメーター
範囲現在のスコープ
構成シリアル化された TPUEmbeddingConfiguration プロト。
戻り値
  • XlaRecvTPUEmbeddingDeduplicationData の新しいインスタンス

public出力<?>出力()