定义:
tf.boolean_mask(
tensor,
mask,
name='boolean_mask',
axis=None
)
参数:
- tensor: N-D tensor.
- mask: K-D boolean tensor, K <= N and K must be known statically.
- name: A name for this operation (optional).
- axis: A 0-D int Tensor representing the axis in tensor to mask from. By default, axis is 0 which will mask from the first dimension. Otherwise K + axis <= N.
返回值:
(N-K+1)-dimensional tensor populated by entries in tensor corresponding to True values in mask.
参数b是一个bool型的矩阵(滤波器掩模),和a进行tf.boolean_mask运算后,返回与b中“True”元素同下标的部分。
常用于目标检测中,用来过滤概率值比较低的锚盒。
参考:https://blog.csdn.net/qq_29444571/article/details/84574526