tensorflow::ops::CompareAndBitpack

#include <math_ops.h>

Сравнение значений input с threshold и упаковка полученных битов в uint8.

Краткое описание

Каждое сравнение возвращает булево значение true (если input_value > threshold) или false в противном случае.

Эта операция полезна для Locality-Sensitive-Hashing (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], выходной тензор имеет форму [s0, s1, ..., s_n / 8].

Аргументы:

• scope: Объект Scope
• input: Значения для сравнения с threshold и упаковки битов.
• threshold: Порог для сравнения.

Возвращаемое значение:

• Output: Упакованные по битам результаты сравнений.

Общедоступные атрибуты

Operation operation

::tensorflow::Output output

Общедоступные функции

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

::tensorflow::Node * node() const 

operator::tensorflow::Input() const 

operator::tensorflow::Output() const