RecvTPUEmbeddingActivations

パブリック最終クラスRecvTPUEmbeddingActivations

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

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

パブリックメソッド

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

継承されたメソッド

パブリックメソッド

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

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

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

public Iterator< Operand <Float>> iterator ()

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

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