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]]