以二进制形式打印整数张量

Question

我有一个 tf.int32 类型的张量 我想使用tf.Print，但我需要结果是二进制的。 这甚至可能吗？ 用于调试。

例子：

constant = tf.constant(5)
#magic
tf.Print(constant) # prints 101

Answer 1

你可以使用tf.py_function:

x = tf.placeholder(tf.int32)
bin_op = tf.py_function(lambda dec: bin(int(dec))[2:], [x], tf.string)

bin_op.eval(feed_dict={x: 5})   # '101'

但请注意，tf.py_function 在图中创建了一个节点。所以如果你想打印很多张量，你可以在tf.Print之前用tf.py_function包装它们，但是在循环中这样做可能会导致膨胀。