ValueError：only one element tensors can be converted to Python scalarsC 笔记

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