0
我必須爲張量中不等於-1的所有數字創建一個掩碼。張量流暢的Numpy-like口罩?
在numpy的我會做到這一點:
mask = np.ones_like(my_matrix)
mask[my_matrix == -1] = 0
我怎樣才能實現tensorflow一樣嗎?
我必須爲張量中不等於-1的所有數字創建一個掩碼。張量流暢的Numpy-like口罩?
在numpy的我會做到這一點:
mask = np.ones_like(my_matrix)
mask[my_matrix == -1] = 0
我怎樣才能實現tensorflow一樣嗎?
mask = tf.not_equal(my_matrix,-1)