tf.where(
condition,
x=None,
y=None,
name=None
)
x,y为和tensor相同维度的tensor,将tensor中的true位置元素替换为a中对应位置元素,false的替换为b中对应位置元素。
import tensorflow as tf
import numpy as np
sess=tf.Session()
a=np.array([[1,0,0],[0,1,1]])
a1=np.array([[3,2,3],[4,5,6]])
print(sess.run(tf.equal(a,1)))
print(sess.run(tf.where(tf.equal(a,1),a1,1-a1)))
[[ True False False]
[False True True]]
[[ 3 -1 -2]
[-3 5 6]]
如果你已经来到这里,不妨关注如下公众号一起探索这个奇妙世界:
![](https://img-blog.csdnimg.cn/20210208230013730.jpg)