ComputeDedupDataTupleMask

パブリック最終クラスComputeDedupDataTupleMask

演算は、埋め込みコアからの重複排除データのタプル マスクを計算します。

エンベディング コアから受信する重複排除データは、type=DT_VARIANT の Tensor です。テンソル自体は XLA ネストされたタプルであり、その要素はランク 1 テンソルです。この演算は、これらの要素のタイプと長さを表すためのものです。

パブリックメソッド

出力<整数>
asOutput ()
テンソルのシンボリック ハンドルを返します。
静的ComputeDedupDataTupleMask
create (スコープスコープ、文字列構成)
新しい 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 番目の次元は、各タプル要素の長さを示します。