tf.where使用

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中对应元素
 

参考文档

tenflow 入门 tf.where(）用法_飞奔的帅帅的博客-CSDN博客_tf.where