声明:
- 需要读者对tensorflow和深度学习有一定了解
tf.boolean_mask实现类似numpy数组的mask操作
Python的numpy array可以使用boolean类型的数组作为索引,获得numpy array中对应boolean值为True的项。示例如下:
# numpy array中的boolean mask
import numpy as np
target_arr = np.arange(5)
print "numpy array before being masked:"
print target_arr
mask_arr = [True, False, True, False, False]
masked_arr = target_arr[mask_arr]
print "numpy array after being masked:"
print masked_arr
运行结果如下:
numpy array before being masked:
[0 1 2 3 4]
numpy array after being masked:
[0 2]
tf.boolean_maks
对目标tensor实现同上述numpy array一样的mask操作,该函数的参数也比较简单,如下所示:
tf.boolean_mask(
tensor, # target tensor
mask, # mask tensor
axis=None,
name='boolean_mask'
)