1、tf.where使用
tf.where(condition, x=None, y=None, name=None)
2、where(condition)的用法
condition是bool型值,True/False
返回值,是condition中元素为True对应的索引
3、where(condition, x=None, y=None)的用法
condition, x, y 相同维度,condition是bool型值,True/False
返回值是对应元素,condition中元素为True的元素替换为x中的元素,为False的元素替换为y中对应元素
参考文档