RecvTPUEmbeddingActivations

パブリック最終クラスRecvTPUEmbeddingActivations

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

TPU システムは、TPUEmbeddingEnqueue(Integer/Sparse/SparseTensor)Batch の引数で指定された埋め込み検索と集計を実行します。これらの集計の結果は、RecvTPUEmbeddingActivations 演算の出力として Tensorflow グラフに表示されます。この演算は、モデルで指定されたテーブルごとに 1 つのアクティベーションの Tensor を含むリストを返します。 TPU グラフには、RecvTPUEmbeddingActivations 演算子が 1 つだけ存在できます。

定数

OP_NAME TensorFlow コア エンジンによって認識される、この演算の名前

パブリックメソッド

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

継承されたメソッド

定数

パブリック静的最終文字列OP_NAME

TensorFlow コア エンジンによって認識される、この演算の名前

定数値: "RecvTPUEmbeddingActivations"

パブリックメソッド

public static RecvTPUEmbeddingActivations create (スコープscope、Long numOutputs、String config)

新しい RecvTPUEmbeddingActivations オペレーションをラップするクラスを作成するためのファクトリ メソッド。

パラメーター
範囲現在のスコープ
出力数出力アクティベーション テンソルの数。モデル内の埋め込みテーブルの数に等しい。
構成シリアル化された TPUEmbeddingConfiguration プロト。
戻り値
  • RecvTPUEmbeddingActivations の新しいインスタンス

public Iterator< Operand < TFloat32 >> iterator ()

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

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