tf.boolean_mask()用法

定义：

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