一、
tf.where
tf.where(condition, x=None, y=None,name=None)
-
condition
:一个Tensor
,数据类型为bool
类型 -
如果
x、y
均为空,那么返回condition
中值为True
的位置 -
如果
x、y
不为空,那么x、y
必须有相同的形状。如果x、y
是标量,那么condition
也必须是标量。如果x、y
是向量,那么condition
必须和x
的某