问题出在交叉熵函数的参数设置上
如:
criterion = nn.CrossEntropyLoss()#使用交叉熵 loss = criterion(x, lables)
参数x和lables的设置出错了
错误原因一:
x必须是二维即以上,lables则为一维。
tensor类型转换维度用reshape函数
参考博客:
错误原因二:
x不是样本对各个类别的概率举证,而设置成了具体的类别下标,导致出错。
参考博客:
我的错误是一,修改代码如下:
x = x.reshape(1,5).requires_grad_(True)