ML Topluluk Günü 9 Kasım! TensorFlow, JAX güncellemeler için bize katılın ve daha fazla bilgi edinin

tensorflow :: ops :: CompareAndBitpack

#include <math_ops.h>

input değerlerini threshold ve ortaya çıkan bitleri bir uint8 .

Özet

Her karşılaştırma bir boolean true ( input_value > threshold ) veya aksi takdirde false döndürür.

Bu işlem, Konum Duyarlı Karma (LSH) ve kosinüs ve L2 mesafelerinin karma yaklaşımlarını kullanan diğer algoritmalar için yararlıdır; kodlar bir girişten şu şekilde oluşturulabilir:

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]

NOT : Şu anda, tensörün en içteki boyutu 8'e bölünebilir olmalıdır.

[s0, s1, ..., s_n] şeklinde bir input verildiğinde, çıktı uint8 tensör şeklinde [s0, s1, ..., s_n / 8] .

Argümanlar:

  • kapsam: Bir Scope nesnesi
  • input: threshold ve bit paketiyle karşılaştırılacak threshold .
  • eşik: Karşılaştırılacak eşik.

İadeler:

  • Output : Bit paketli karşılaştırmalar.

Yapıcılar ve Yıkıcılar

CompareAndBitpack (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input threshold)

Genel özellikler

operation
output

Kamusal işlevler

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Genel özellikler

operasyon

Operation operation

çıktı

::tensorflow::Output output

Kamusal işlevler

CompareAndBitpack

 CompareAndBitpack(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input threshold
)

düğüm

::tensorflow::Node * node() const 

operator :: tensorflow :: Girdi

 operator::tensorflow::Input() const 

operator :: tensorflow :: Çıktı

 operator::tensorflow::Output() const