解决RuntimeError: Dimension out of range (expected to be in range of [-1, 0], but got 1)
问题描述:x=[[0.1,0.2,0.1],[0.4,0.1,0.3]]y=[1,0]loss_fn=nn.CrossEntropyLoss()loss=loss_fn(x,y)x为特征矩阵,y为标签。x对应维度应为batch_sizexdim。y对应维度应为batch_size。当使用交叉熵时,当最后一个batch_size为1时。造成x维度变成1,y也为1。此时nn.CrossEntropyL