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]
相關文章
相關標籤/搜索