找出tensor里所有True值的index
import tensorflow as tf
a = tf.constant([False,False,True,False,True],dtype=tf.bool)
b = tf.where(tf.equal(a,True))
sess = tf.Session()
print(sess.run(b))
print:
[[2]
[4]]
找出tensor里所有True值的index
import tensorflow as tf
a = tf.constant([False,False,True,False,True],dtype=tf.bool)
b = tf.where(tf.equal(a,True))
sess = tf.Session()
print(sess.run(b))
print:
[[2]
[4]]