IndexError: Dimension out of range (expected to be in range of [-1, 0], but got 1)

报错代码

criterion = nn.CrossEntropyLoss().to(device) loss = criterion(output, target)

网上很多说criterion的第一个输入问题， 我的解决是改变batchsize 大小即可