返回输入矩阵中true的索引位置,x和y必须同时有(维度必须相同)或没有
import tensorflow as tf
where = tf.where([True,False,True,False,True,True])
where1 = tf.where([True,False,True,False,True,True],x=[1.,0.2,0.5,0.,2.,3.],y=[1,2,3,4,5,6])
with tf.Session() as sess:
print(sess.run(where))
print(sess.run(where1))
[[0]
[2]
[4]
[5]]
[1. 2. 0.5 4. 2. 3. ]