标签： tensorflow binary int
我想将整数张量a_dec=tf.constant([2,7], dtype=tf.int32)转换为二进制秩-2张量，例如：a_bin=tf.constant([[0,1,0],[1,1,1]], dtype=tf.int32)。
a_dec=tf.constant([2,7], dtype=tf.int32)
a_bin=tf.constant([[0,1,0],[1,1,1]], dtype=tf.int32)
有没有有效的方法呢？
答案 0 :(得分：1)
如果您知道矢量的大小n，那么：
n
a_bin = tf.mod(tf.bitwise.right_shift(tf.expand_dims(a_dec,1), tf.range(n)), 2)