Pytorch之CrossEntropyLoss() 与 NLLLoss() 的区别_基本语法



  • NLLLoss 的 输入 是一个对数概率向量和一个目标标签(不需要是one-hot编码形式的). 它不会为我们计算对数概率. 适合网络的最后一层是log_softmax. 损失函数 nn.CrossEntropyLoss() 与 NLLLoss() 相同, 唯一的不同是它为我们去做 softmax.

Pytorch之CrossEntropyLoss() 与 NLLLoss() 的区别_代价函数_02



C/C++基本语法学习 STL C++ primer