- 博客(1)
- 收藏
- 关注
RSS订阅原创 使用pytorch获取tensor每行中的top k
使用pytorch获取tensor每行中的top k 最近刚刚入门pytorch，测试完模型计算准确率需要R@1，R@5等，想获取一下tensor每行前k个大的数，奈何搜了一圈就没说到点子上的…参照numpy数组，使用了一下tensor.argsort(),成功。 ...
2019-10-23 21:52:06
4095
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人