参考链接
https://blog.youkuaiyun.com/ustbbsy/article/details/79564828
用法
where(condition, x=None, y=None, name=None)
注意:
- condition, x, y 相同维度
- condition是bool型值,True/False
保留a中,condition中true的位置
其余位置用b的对应位置去填
示例代码
import tensorflow as tf
a = [[1,2,3],[4,5,6]]
b = [[1,0,3],[1,5,1]]
condition1 = [[True,False,False],
[False,True,True]]
condition2 = [[True,False,False],
[False,True,False]]
aa = tf.where(condition1,a,b)
bb = tf.where(condition2,a,b)
得到
aa

bb

文章介绍了在TensorFlow中使用tf.where函数根据条件选择保留数组元素的方法。示例代码展示了如何根据二维bool数组condition,从a和b两个矩阵中选择值,True部分保留a的值,False部分取b的值。
4976





