如果是二分类问题,可以用 nn.BCELoss() 或者 nn.BCEWithLogitsLoss(),区别是后者集成了sigmoid层。也就是BCEWithLogitsLoss() = sigmoid() + BCELoss() nn.CrossEntropyLoss() 输入需要注意以下几点: nn.CrossEntropyLoss() 内置了softmax操作,因此input只需要是网络输出的logits即可,不需要自己用softmax进行归...