1. 首页
  2. 问答
  3. 张量流畅的Numpy-like口罩？

张量流畅的Numpy-like口罩？

2017-09-24 17 views
0

我必须为张量中不等于-1的所有数字创建一个掩码。张量流畅的Numpy-like口罩？

在numpy的我会做到这一点：

mask = np.ones_like(my_matrix) 
mask[my_matrix == -1] = 0

我怎样才能实现tensorflow一样吗？

来源

2017-09-24 AFusco

回答

1

mask = tf.not_equal（my_matrix，-1）

来源

2017-09-24 23:19:14 ppwwyyxx

相关问题

 相关问题