我正在執(zhí)行類似 MNIST 的任務，輸入是 10 類圖像，預期輸出是圖像的預測類。但現(xiàn)在output就像 [-2.3274, -2.2723, ...]，其長度是 batch_size。并且target是 [4., 2., 2., 8., ...]Error message: RuntimeError: 標量類型 Long 的預期對象，但參數(shù) #2 'target' 的標量類型浮點數(shù)class Net(nn.Module):    ...    ...    def forward(self, x):        ...        ...        return F.log_softmax(x, dim = 1)criterion = torch.nn.NLLLoss()誰能給我一些建議？謝謝。