XlaRecvTPUEmbeddingActivations

パブリック最終クラスXlaRecvTPUEmbeddingActivations

TPU 上で埋め込みアクティベーションを受け取る操作。

TPU システムは、埋め込み検索と集計を実行します。これらの集計の結果は、XlaRecvTPUEmbeddingActivations Op の出力として Tensorflow グラフに表示されます。この演算は、モデルで指定されたテーブルごとに 1 つのアクティベーションの Tensor を含むリストを返します。

パブリックメソッド

静的XlaRecvTPUEmbeddingActivations
create (スコープscope、オペランド<?> deduplicationData、Long numTables、String config)
新しい XlaRecvTPUEmbeddingActivations 操作をラップするクラスを作成するファクトリ メソッド。
イテレータ<オペランド< 浮動小数点 >>
イテレータ()
リスト<出力<浮動小数点数>>
出力()
モデル内の埋め込みテーブルごとに 1 つの Tensor を含む埋め込みアクティベーションの TensorList。

継承されたメソッド

パブリックメソッド

public static XlaRecvTPUEmbeddingActivations create (スコープscope、オペランド<?> deduplicationData、Long numTables、String config)

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

パラメーター
範囲現在のスコープ
重複排除データ重複排除データを含む type=DT_VARIANT の Tensor。テンソルは、N 個の要素を含む XLA ネストされたタプルです (N は、TPU チップあたりのテンソル コアに対する埋め込みの数の比率です)。入れ子になったタプルの各要素は、ランク 1 テンソルのタプルです。各テンソルには、TensorCore での埋め込みルックアップ用のインデックス (DT_UINT32) または埋め込みルックアップ操作の出力に適用する重み (DT_FLOAT) が含まれています。
テーブル数出力アクティベーションテンソルの数。機能記述子が TPU 埋め込み構成に存在する場合、それは機能の数と等しく、それ以外の場合はモデル内の埋め込みテーブルの数と等しくなります。
構成シリアル化された TPUEmbeddingConfiguration プロト。
戻り値
  • XlaRecvTPUEmbeddingActivations の新しいインスタンス

public Iterator< Operand <Float>> iterator ()

public List< Output <Float>>出力()

モデル内の埋め込みテーブルごとに 1 つの Tensor を含む埋め込みアクティベーションの TensorList。