tensorflow 中 boolean_mask的具体含义和例程解析

tensorflow里的一个函数，boolean_mask(a,b) 将使a (m维)矩阵仅保留与b中“True”元素同下标的部分。在做目标检测（YOLO）时常常用到。 """Apply boolean mask to tensor.  Numpy equivalent is `tensor[mask]`. ```python # 1-D example tensor = [0, 1, 2, 3]

>>阅读原文<<