invalid argument 5: k not in range for dimension at /opt/conda/conda-bld/pytorch_1607370172916/work/aten/src/THC/generic/THCTensorTopK.cu:26

报错原因：我训练的模型是二分类，所以根本没有五个类

 

torch.topk(input, k, dim=None, largest=True, sorted=True, out=None) -> (Tensor, LongTensor)

input：一个tensor数据
k：指明是得到前k个数据以及其index
dim： 指定在哪个维度上排序， 默认是最后一个维度
largest：如果为True，按照大到小排序； 如果为False，按照小到大排序
sorted：返回的结果按照顺序返回
out：可缺省，不要

 

posted @   Tomorrow1126  阅读(438)  评论(0)  编辑  收藏  举报
相关博文：
· torch topk函数
· pytorch view函数
· torch topk 使用
· pytorch报错IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python
· torch.max
阅读排行：
· 阿里最新开源QwQ-32B，效果媲美deepseek-r1满血版，部署成本又又又降低了！
· SQL Server 2025 AI相关能力初探
· 单线程的Redis速度为什么快？
· AI编程工具终极对决：字节Trae VS Cursor，谁才是开发者新宠？
· 开源Multi-agent AI智能体框架aevatar.ai，欢迎大家贡献代码
历史上的今天：
2020-03-25 手写数字识别
2020-03-25 一个讲pyYaml的博客
2020-03-25 tf.cast()
2020-03-25 argparse基本用法
点击右上角即可分享
微信分享提示
AI IDE Trae