使用pytorch获取tensor每行中的top k 最近刚刚入门pytorch,测试完模型计算准确率需要R@1,R@5等,想获取一下tensor每行前k个大的数,奈何搜了一圈就没说到点子上的…参照numpy数组,使用了一下tensor.argsort(),成功。