torch.nn.CrossEntropyLoss()

最新推荐文章于 2024-08-23 16:49:02 发布

一壶浊酒..

最新推荐文章于 2024-08-23 16:49:02 发布

阅读量567

点赞数 4

分类专栏： # pytorch 文章标签：深度学习人工智能

版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

本文链接：https://blog.csdn.net/qq_40107571/article/details/135469911

版权

pytorch 专栏收录该内容

57 篇文章 5 订阅 ¥39.90 ¥99.00

订阅专栏

nn.CrossEntropyLoss()是PyTorch用于多分类问题的损失函数，结合了LogSoftmax和NLLLoss。它接受输入张量(input)和目标张量(target)，其中input是未经softmax的原始输出，target是样本的真实标签。可选参数包括weight指定类别权重，ignore_index忽略特定类别，reduction控制损失计算方式，默认为'sum'。另外，label_smoothing可用于标签平滑，帮助模型泛化。

摘要由CSDN通过智能技术生成

torch.nn.CrossEntropyLoss(weight=None,size_average=None,ignore_index=-100,reduce=None,reduction='mean',label_smoothing=0.0)

计算过程

nn.CrossEntropyLoss()=nn.LogSoftmax()+nn.NLLLoss()

import torch
import torch.nn as nn

loss_func = nn.CrossEntropyLoss()
pre = torch.tensor([0.8, 0.5, 0.2, 0.5], dtype=torch.float)
tgt = torch.tensor([1, 0, 0, 0], dtype=torch.float)
print("手动计算:")
print("1.softmax")
print(torch.softmax(pre, dim=-1))
print("2.取对数")
print(torch.log(torch.softmax(pre, dim=-1)))
print("3.与真实值相乘")
print(-torch.sum(torch.mul(torch.log(torch.softmax(pre, dim=-1)), tgt), dim=-1))
pri

了解本专栏

关注

4
点赞
踩
6

收藏

觉得还不错? 一键收藏
打赏
0
评论
复制链接

分享到 QQ

分享到新浪微博

扫一扫

专栏目录

一壶浊酒.. CSDN认证博客专家 CSDN认证企业博客

码龄7年

468: 原创

1万+: 周排名

7111: 总排名

47万+: 访问

: 等级

5368: 积分

1481: 粉丝

612: 获赞

74: 评论

1494: 收藏

私信

关注

热门文章

分类专栏

最新评论

python tkinter 简介
失心疯_2023: 博主文章写的非常详细，图文并茂，优质好文！学到了很多新的知识点，感谢博主分享！ ============================================ 👉👉 Python零基础入门教程合集 👈👈：https://blog.csdn.net/weixin_50296259/article/details/131907622 👉👉 Python项目虚拟环境(超详细讲解) 👈👈：https://blog.csdn.net/weixin_50296259/article/details/142168781 👉👉 Python GUI(PyQt5)文章合集 👈👈：https://blog.csdn.net/weixin_50296259/article/details/130674350 👉👉 Oracle数据库文章合集 👈👈：https://blog.csdn.net/weixin_50296259/article/details/130739064 👉👉 优质资源下载合集 👈👈：https://blog.csdn.net/weixin_50296259/article/details/130674430 👉👉 Python爬虫从入门到入狱系列合集 👈👈：https://blog.csdn.net/weixin_50296259/article/details/135698886
python tkinter 简介
失心疯_2023: 博主文章写的非常详细，图文并茂，优质好文！学到了很多新的知识点，感谢博主分享！ ============================================ 👉👉 Python零基础入门教程合集 👈👈：https://blog.csdn.net/weixin_50296259/article/details/131907622 👉👉 Python项目虚拟环境(超详细讲解) 👈👈：https://blog.csdn.net/weixin_50296259/article/details/142168781 👉👉 Python GUI(PyQt5)文章合集 👈👈：https://blog.csdn.net/weixin_50296259/article/details/130674350 👉👉 Oracle数据库文章合集 👈👈：https://blog.csdn.net/weixin_50296259/article/details/130739064 👉👉 优质资源下载合集 👈👈：https://blog.csdn.net/weixin_50296259/article/details/130674430 👉👉 Python爬虫从入门到入狱系列合集 👈👈：https://blog.csdn.net/weixin_50296259/article/details/135698886
手把手教你学51单片机-点亮你的LED
Raoff: 为什么我不能成功点亮
Python logging模块的基本使用
北风之神c: 总结的很全面，写得赞，博主用心了。此国产日志 https://nb-log-doc.readthedocs.io/zh_CN/latest 使用原生 loggng封装，兼容性和替换性100%,大幅简化logging的使用。 1、日志能根据级别能够自动变彩色。 1 2、print自动变彩色。 3、日志和print在pycahrm控制台的输出都自动可以点击跳转到文件和行号。 4、多进程日志切割安全，文件日志写入性能高。 5、入参简单，能一键自动记录到多种地方。 6、 nb_log 兼容包含loguru色彩模式,loguru只是nb_log的子集之一. 相比 loguru 有10胜。 pip install nb_log 。
python Canny边缘检测
白社会社长: 第二步好像没有用到sobel算子，这是为什么呢？

大家在看

最新文章

目录

评论

被折叠的条评论为什么被折叠?

到【灌水乐园】发言

查看更多评论

添加红包

成就一亿技术人!

hope_wisdom

发出的红包

打赏作者

一壶浊酒.. 你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20

扫码支付：¥1

获取中

扫码支付

您的余额不足，请更换扫码支付或充值

实付元

使用余额支付

点击重新获取

扫码支付

钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。