tf.where()定义如下:

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

condition:一个Tensor,数据类型为tf.bool类型

如果x、y均为空,那么返回condition中值为True的位置的Tensor:例如:x就是condition,y是返回值

tf.where()函数

x中值为True的位置有[0,0,0],[0,1,1],[0,2,0],[0,2,1],[1,0,1],[1,1,0],[1,2,0]。

如果x、y不为空,那么x、y必须有相同的形状。如果x、y是标量,那么condition参数也必须是标量。如果x、y是向量,那么condition必须和x的第一维有相同的形状或者和x形状一致。

返回值:如果x、y不为空的话,返回值和x、y有相同的形状,如果condition对应位置值为True那么返回Tensor对应位置为x的值,否则为y的值.  例子:

tf.where()函数


相关文章:

  • 2022-12-23
  • 2021-08-18
  • 2021-11-05
  • 2021-11-26
  • 2021-12-20
  • 2022-12-23
  • 2021-04-05
  • 2022-12-23
猜你喜欢
  • 2021-05-17
  • 2021-09-07
  • 2022-12-23
  • 2021-06-30
  • 2021-10-18
  • 2021-10-27
相关资源
相似解决方案