从Tensorflow中的多热向量中提取索引

Question

如何从Tensorflow中的多热张量中提取索引？

例如

a = (1, 1, 0, 1, 0, 0, 0)

然后，输出应为

output = (0, 1, 3)

Answer 1

如果你是这个意思

a = tf.Variable(initial_value = [1, 1, 0, 1, 0, 0, 0])
print( sess.run( tf.where(tf.equal(a, tf.constant(1)) )))

那么输出就是这个。

  

[[0]     [1]     [3]]