标签： python numpy tensorflow
我必须为张量中不等于-1的所有数字创建一个掩码。
在numpy我会这样做：
mask = np.ones_like(my_matrix) mask[my_matrix == -1] = 0
如何在tensorflow中实现相同的目标？
答案 0 :(得分：1)
mask = tf.not_equal（my_matrix，-1）