声明:
- 需要读者对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'
)
下面,我们来尝试一下

本文介绍了TensorFlow中的masking和padding操作,包括`tf.boolean_mask`用于实现类似numpy的mask操作,`tf.ragged.boolean_mask`处理元素维度的mask,以及`padded_batch`函数用于在构建batch时自动填充不同长度的序列。内容涵盖示例和原理解释。
最低0.47元/天 解锁文章
1万+

被折叠的 条评论
为什么被折叠?



