参考链接
https://blog.csdn.net/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