tf.boolean_mask(a, mask)
函数将使a
矩阵仅保留与mask
中True
元素同下标的部分,即通过mask
提取数据。
import tensorflow as tf
mask = [True, False, True, True]
a = [1,2,3,4]
tf.boolean_mask(a, mask)
>>> array([1, 3, 4])
tf.boolean_mask(a, mask)
函数将使a
矩阵仅保留与mask
中True
元素同下标的部分,即通过mask
提取数据。
import tensorflow as tf
mask = [True, False, True, True]
a = [1,2,3,4]
tf.boolean_mask(a, mask)
>>> array([1, 3, 4])