パブリック最終クラスRecvTPUEmbeddingActivations
TPU 上で埋め込みアクティベーションを受け取る操作。
TPU システムは、TPUEmbeddingEnqueue(Integer/Sparse/SparseTensor)Batch の引数で指定された埋め込み検索と集計を実行します。これらの集計の結果は、RecvTPUEmbeddingActivations 演算の出力として Tensorflow グラフに表示されます。この演算は、モデルで指定されたテーブルごとに 1 つのアクティベーションの Tensor を含むリストを返します。 TPU グラフには、RecvTPUEmbeddingActivations 演算子が 1 つだけ存在できます。
定数
弦 | OP_NAME | TensorFlow コア エンジンによって認識される、この演算の名前 |
パブリックメソッド
静的RecvTPUEmbeddingActivations | |
イテレータ<オペランド< TFloat32 >> | イテレータ() |
リスト<出力< TFloat32 >> | 出力() モデル内の埋め込みテーブルごとに 1 つの Tensor を含む埋め込みアクティベーションの TensorList。 |
継承されたメソッド
定数
パブリック静的最終文字列OP_NAME
TensorFlow コア エンジンによって認識される、この演算の名前
定数値: "RecvTPUEmbeddingActivations"
パブリックメソッド
public static RecvTPUEmbeddingActivations create (スコープscope、Long numOutputs、String config)
新しい RecvTPUEmbeddingActivations オペレーションをラップするクラスを作成するためのファクトリ メソッド。
パラメーター
範囲 | 現在のスコープ |
---|---|
出力数 | 出力アクティベーション テンソルの数。モデル内の埋め込みテーブルの数に等しい。 |
構成 | シリアル化された TPUEmbeddingConfiguration プロト。 |
戻り値
- RecvTPUEmbeddingActivations の新しいインスタンス