2020年5月28日
pytorch ：使用两次sort函数（排序）找出矩阵每个元素在升序或降序排列中的位置
摘要： 在SSD的代码中经常有见到如下的操作： _, idx = flt[:, :, 0].sort(1, descending=True)#大小为[batch size, num_classes*top_k] _, rank = idx.sort(1)#再对索引升序排列，得到其索引作为排名rank 其作用 阅读全文
posted @ 2020-05-28 18:16 慢行厚积 阅读(3291) 评论(0) 推荐(1) 编辑