tf.boolean_mask(A,B)
其中B的元素只有 Ture 和 False , 该函数返回的是 A中 与B中Ture元素下标 相同的元素
如:
A=[1,2,3,4]
B=[ Ture ,False, Ture, False]
C=tf.boolean_mask(A,B)
输出为:
C=[1,3]
tf.boolean_mask(A,B)
其中B的元素只有 Ture 和 False , 该函数返回的是 A中 与B中Ture元素下标 相同的元素
如:
A=[1,2,3,4]
B=[ Ture ,False, Ture, False]
C=tf.boolean_mask(A,B)
输出为:
C=[1,3]