tensorflow之tf.where实例解析

 

tf.where(
	condition,
	x=None,
	y=None,
	name=None
)

x,y为和ｔｅｎｓｏｒ相同维度的ｔｅｎｓｏｒ，将ｔｅｎｓｏｒ中的ｔｒｕｅ位置元素替换为ａ中对应位置元素，ｆａｌｓｅ的替换为ｂ中对应位置元素。

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]]

 

如果你已经来到这里，不妨关注如下公众号一起探索这个奇妙世界：

有酒有风