パブリック最終クラスComputeDedupDataTupleMask
演算は、埋め込みコアからの重複排除データのタプル マスクを計算します。
エンベディング コアから受信する重複排除データは、type=DT_VARIANT の Tensor です。テンソル自体は XLA ネストされたタプルであり、その要素はランク 1 テンソルです。この演算は、これらの要素のタイプと長さを表すためのものです。
パブリックメソッド
| 出力<整数> | asOutput () テンソルのシンボリック ハンドルを返します。 |
| 静的ComputeDedupDataTupleMask | |
| 出力<整数> | 出力形状() 2 次元 int テンソルは、「XlaRecvTPUEmbeddingDeduplicationData」によって生成される重複排除データ タプルのマスクを表します。 |
継承されたメソッド
パブリックメソッド
public Output <整数> asOutput ()
テンソルのシンボリック ハンドルを返します。
TensorFlow オペレーションへの入力は、別の TensorFlow オペレーションの出力です。このメソッドは、入力の計算を表すシンボリック ハンドルを取得するために使用されます。
public static ComputeDedupDataTupleMask create (スコープスコープ、文字列構成)
新しい ComputeDedupDataTupleMask オペレーションをラップするクラスを作成するファクトリ メソッド。
パラメーター
| 範囲 | 現在のスコープ |
|---|---|
| 構成 | シリアル化された TPUEmbeddingConfiguration プロト。 |
戻り値
- ComputeDedupDataTupleMask の新しいインスタンス
public Output <Integer> OutputShape ()
2 次元 int テンソルは、「XlaRecvTPUEmbeddingDeduplicationData」によって生成される重複排除データ タプルのマスクを表します。タプルには、整数および浮動小数点型の 1 次元テンソル タプル要素がいくつかあります。この Output_shape 2 次元テンソルの最初の次元は、タプル要素のテンソル タイプであり、「0」は整数テンソルを表し、「1」は浮動小数点テンソルを表します。 `output_shape` の 2 番目の次元は、各タプル要素の長さを示します。