index = int(output.argmax(1)) print(index)
我的代码是0~9数字识别 有10个类别，出错的问题，主要是argmax使用错误，原因是输入图像为4通道的，r g b α通道 ，把图像转为灰度图像即1个通道输入网络就能解决问题，
深入分析：output=tensor([[...........]])
index = int(output.argmax(1)) print(index)
图片（尺寸4*32*32）的模型预测输出值output 为tensor张量，尺寸为4*10，是因为图片有四个通道，output.argmax（1） 返回的是一个1*4的tensor（可以认为是数组），那肯定不能int一个数组啊！把图像由4通道变成1通道，output 则变成1*10的tensor ，然后output.argmax（1）返回的1*1，一个元素，则可以int ，返回index
关于argmax的返回值如下
argmax（axis）
tensor 2*3 axis=0 返回 1*3
tensor 2*3 axis=1 返回 1*2
