tf.where的使用方法

tf.where( condition, x=None, y=None, name=None )

1、x=None, y=None时，condition是bool型值，True/False

返回值，是condition中元素为True对应的索引

2、condition， x, y 相同维度，condition是bool型值，True/False

返回值是对应元素，condition中元素为True的元素替换为x中的元素，为False的元素替换为y中对应元素。

a=tf.Variable([[2,0],[1,-1]])
init=tf.global_variables_initializer()
sess.run(init)
b=tf.where(tf.greater(a,1),tf.ones_like(a),tf.zeros_like(a))
print(sess.run(b))

返回：

[[1 0]
 [0 0]]