2017-09-24 48 views
0

我必須爲張量中不等於-1的所有數字創建一個掩碼。張量流暢的Numpy-like口罩?

在numpy的我會做到這一點:

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

我怎樣才能實現tensorflow一樣嗎?

回答

1

mask = tf.not_equal(my_matrix,-1)