tf.where的用法

用法一：

tf.where(bool, a, b)
#bool,a, b 都为tensor张量，bool为真返回a，为假返回b

用法二：

tf.where(tensor, a, b)
#a,b和tensor为相同维度的tensor张量，将tensor中的true位置元素替换为a中对应位置元素，false的替换为b中对应位置元素。