PyTorch的NLLLoss和CrossEntropyLoss区别和联系

最新推荐文章于 2025-03-14 18:09:14 发布

蛐蛐蛐

最新推荐文章于 2025-03-14 18:09:14 发布

阅读量186

点赞数

分类专栏：科研工具深度学习 Python技巧

本文链接：https://blog.youkuaiyun.com/qysh123/article/details/119734714

版权

科研工具同时被 3 个专栏收录

137 篇文章

订阅专栏

Python技巧

99 篇文章

订阅专栏

深度学习

65 篇文章

订阅专栏

这个问题其他网友已经总结得很好，例如：https://blog.youkuaiyun.com/qq_22210253/article/details/85229988

为了方便大家更进一步理解，我把上面博客中的例子总结到了一个文件中，方便大家以此为例学习：

import torch
import torch.nn as nn

input=torch.randn(3,3)
print(input)

sm=nn.Softmax(dim=1)
print(sm(input))

log_result=torch.log(sm(input))
print(log_result)

###############################
loss=nn.NLLLoss()
target=torch.tensor([0,2,1])
input_sm=sm(input)
print(loss(torch.log(input_sm),target))
###############################
loss=nn.CrossEntropyLoss()
print(loss(input,target))