パブリック最終クラスTPUEmbeddingActivations
TPU 埋め込みの差別化を可能にするオペレーション。
この演算は単に最初の入力を返します。これは、TPUEmbeddingDequeueActivations によって返された Tensor からスライスされたものと想定されます。この演算の存在と、その最初の引数がトレーニング可能な変数であることにより、TPU 埋め込み Python ライブラリを介して埋め込みを含むグラフの自動微分が可能になります。
パブリックメソッド
| 出力<浮動小数点> | asOutput () テンソルのシンボリック ハンドルを返します。 |
| 静的TPUEmbeddingActivations | |
| 出力<浮動小数点> | 出力() |
継承されたメソッド
パブリックメソッド
public Output <Float> asOutput ()
テンソルのシンボリック ハンドルを返します。
TensorFlow オペレーションへの入力は、別の TensorFlow オペレーションの出力です。このメソッドは、入力の計算を表すシンボリック ハンドルを取得するために使用されます。
public static TPUEmbeddingActivations create (スコープscope、オペランド<Float>embeddingVariable、オペランド<Float>slicedActivations、Long tableId、Long lookupId)
新しい TPUEmbeddingActivations 操作をラップするクラスを作成するためのファクトリ メソッド。
パラメーター
| 範囲 | 現在のスコープ |
|---|---|
| 埋め込み変数 | トレーニング可能な変数。オプティマイザーがこの演算を見つけられるようにします。 |
| スライスされたアクティベーション | 埋め込みにより Tensor が返されるようにアクティブ化されます。 |
| テーブルID | これらのアクティベーションが計算された埋め込み層構成内のテーブルの ID。 |
| ルックアップID | これらのアクティベーションを生成した埋め込みインデックスのセットの識別子。 |
戻り値
- TPUEmbeddingActivations の新しいインスタンス