CompareAndBitpack

Comparez les valeurs de « input » à « threshold » et regroupez les bits résultants dans un « uint8 ».

Chaque comparaison renvoie un booléen « vrai » (si « valeur_entrée > seuil ») ou et « faux » sinon.

Cette opération est utile pour le Locality-Sensitive-Hashing (LSH) et d'autres algorithmes qui utilisent des approximations de hachage des distances cosinus et « L2 » ; les codes peuvent être générés à partir d'une entrée via :

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]
 

Étant donné une « entrée » en forme de « [s0, s1, ..., s_n] », la sortie est un tenseur « uint8 » en forme de « [s0, s1, ..., s_n / 8] ».