会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
慢行厚积
博客园
首页
新随笔
联系
订阅
管理
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)
编辑
公告