【pytorch】nn.CrossEntropyLoss()和nn.NLLLoss()的区别

损失函数NLLLoss() 的 输入 是一个对数概率向量和一个目标标签. 它不会为我们计算对数概率，适合最后一层是log_softmax()的网络. 损失函数 CrossEntropyLoss() 与 NLLLoss() 类似, 唯一的不同是它为我们去做 softmax.可以理解为：

CrossEntropyLoss()=log_softmax() + NLLLoss() 

参考：

https://blog.csdn.net/hao5335156/article/details/80607732