例如,
a = (1, 1, 0, 1, 0, 0, 0)
那么,输出应该是
output = (0, 1, 3)
如果你是这个意思
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]