יום הקהילה ML הוא 9 בנובמבר! הצטרפו אלינו עדכונים מ- TensorFlow, JAX, ועוד למידע נוסף

זרימת טנסור :: אופ :: השווה AndBitpack

#include <math_ops.h>

השווה בין ערכי input threshold וארוז ביטים שנוצרו לתוך uint8 .

סיכום

כל השוואה מחזירה true בוליאנית (אם input_value > threshold ) או false אחרת.

פעולה זו שימושית עבור Locality-Sensitive-Hashing (LSH) ואלגוריתמים אחרים המשתמשים בקירוב hashing של מרחקי קוסינוס ו- 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] uint8 [s0, s1, ..., s_n / 8] .

טענות:

  • היקף: אובייקט Scope
  • קלט: ערכים להשוואה מול threshold ו- bitpack.
  • סף: סף להשוואה נגד.

החזרות:

  • Output : ההשוואות הסיביות.

קונסטרוקטורים ומשחתנים

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

תכונות ציבוריות

operation
output

פונקציות ציבוריות

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

תכונות ציבוריות

מבצע

Operation operation

תְפוּקָה

::tensorflow::Output output

פונקציות ציבוריות

השווה AndBitpack

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

צוֹמֶת

::tensorflow::Node * node() const 

אופרטור :: זורם טנסור :: קלט

 operator::tensorflow::Input() const 

אופרטור :: זרימת טנסור :: פלט

 operator::tensorflow::Output() const