CompareAndBitpack

パブリック最終クラスCompareAndBitpack

「input」の値を「threshold」と比較し、結果のビットを「uint8」にパックします。

各比較は、ブール値の「true」 (「input_value」 > 「threshold」の場合) を返し、それ以外の場合は「false」を返します。

この操作は、局所性依存ハッシュ (LSH) や、コサインと「L2」距離のハッシュ近似を使用するその他のアルゴリズムに役立ちます。コードは

codebook_size = 50
 codebook_bits = codebook_size * 32
 codebook = tf.get_variable('codebook', [x.shape[-1].value, codebook_bits],
                            dtype=x.dtype,
                            initializer=tf.orthogonal_initializer())
 codes = compare_and_threshold(tf.matmul(x, codebook), threshold=0.)
 codes = tf.bitcast(codes, tf.int32)  # go from uint8 to int32
 # now codes has shape x.shape[:-1] + [codebook_size]
 
を介して入力から生成できます。: 現在、テンソルの最も内側の次元は 8 で割り切れる必要があります。

`input` の形状が `[s0, s1, ..., s_n]` の場合、出力は `uint8` のテンソル形状の `[s0, s1, ..., s_n / 8]` になります。

定数

OP_NAME TensorFlow コア エンジンによって認識される、この演算の名前

パブリックメソッド

出力< TUint8 >
asOutput ()
テンソルのシンボリック ハンドルを返します。
static <T extends TType > CompareAndBitpack
create (スコープスコープ、オペランド<T> 入力、オペランド<T> しきい値)
新しい CompareAndBitpack 操作をラップするクラスを作成するファクトリ メソッド。
出力< TUint8 >
出力()
ビットパックされた比較。

継承されたメソッド

定数

パブリック静的最終文字列OP_NAME

TensorFlow コア エンジンによって認識される、この演算の名前

定数値: "CompareAndBitpack"

パブリックメソッド

public Output < TUint8 > asOutput ()

テンソルのシンボリック ハンドルを返します。

TensorFlow オペレーションへの入力は、別の TensorFlow オペレーションの出力です。このメソッドは、入力の計算を表すシンボリック ハンドルを取得するために使用されます。

public static CompareAndBitpack create (スコープスコープ、オペランド<T> 入力、オペランド<T> しきい値)

新しい CompareAndBitpack 操作をラップするクラスを作成するファクトリ メソッド。

パラメーター
範囲現在のスコープ
入力「しきい値」およびビットパックと比較する値。
しきい値比較するしきい値。
戻り値
  • CompareAndBitpack の新しいインスタンス

public出力< TUint8 >出力()

ビットパックされた比較。