讲师中心 微信公众号
AI工具推荐 视频效率加速

C++怎么计算交叉熵损失函数

梦宇同学_5443

梦宇同学_5443

发布时间:2026-10-20 13:24:29

|

991人浏览过

|

来源于php中文网

原创

C++标准库无交叉熵损失函数,需手写或用第三方库;手写时须用log-sum-exp稳定技巧避免溢出,如先减最大值再计算;libtorch中可用torch::nn::functional::cross_entropy,要求targets为int64_t类型且索引不越界。

c++怎么计算交叉熵损失函数

交叉熵损失在 C++ 里没有标准库函数

标准 C++ 库(<cmath>、<algorithm> 等)不提供 cross_entropy_loss 这类函数。你得自己写,或依赖第三方库(如 Eigen、xtensor、libtorch)。直接手写最可控,也最容易调试。

常见错误是把 logits 当作概率直接喂给 -log(p) ——这会导致数值不稳定甚至 nan,因为未归一化的 logits 可能极大或极小,exp(logits) 溢出,log(0) 崩溃。

  • 必须先对 logits 做 softmax(或更稳的 log-softmax),再计算 loss
  • 分类任务中,label 通常是 one-hot 或 class index,要对应处理
  • 多分类单标签场景最常用:输入 shape 是 [batch_size, num_classes],label 是 [batch_size] 的整数索引

手写稳定版 cross_entropy_loss(单样本 + 批量)

核心是避免 exp 溢出:用 softmax 的 log-space 形式,即 log_softmax,再取负对数似然。公式等价于:

loss = -log(softmax(logits)[target]) = -logits[target] + log(sum(exp(logits)))

但直接算 sum(exp(logits)) 危险,所以改用带减法平移的 trick:

立即学习“C++免费学习笔记(深入)”;

C++14
C++14

C++14 对 C++11 的修正与增强版本,适合旧系统维护和较老工具链兼容。

下载
  • 先求 max_logit = max(logits)
  • 再算 log(sum(exp(logits - max_logit))) + max_logit —— 这就是稳定版的 log(sum(exp(logits)))
  • 最终 loss = -logits[target] + log(sum(exp(logits - max_logit))) + max_logit

示例(单样本,logits = {2.0, 1.0, 0.1}, target = 0):

max_logit = 2.0<br>shifted = {0.0, -1.0, -1.9}<br>log_sum_exp = log(exp(0.0) + exp(-1.0) + exp(-1.9)) + 2.0 ≈ log(1 + 0.3679 + 0.1496) + 2.0 ≈ 2.405<br>loss = -2.0 + 2.405 = 0.405

用 libtorch(C++ API)时怎么调用

如果你已在用 PyTorch 训练流程,C++ 部署时推荐直接用 torch::nn::functional::cross_entropy,它默认启用 log-sum-exp 稳定实现,且支持 label 为 int64_t 索引(无需 one-hot)。

  • 输入 logits 是 torch::Tensor,shape [N, C]
  • targets 是 torch::Tensor,dtype torch::kLong,shape [N]
  • 别漏掉 reduction="mean"(默认)或设为 "none" 得到 per-sample loss
  • 错误示例:targets 传 float 类型 → 报错 expected scalar type Long but found Float

代码片段:

auto logits = torch::randn({32, 10});<br>auto targets = torch::randint(0, 10, {32}, torch::kLong);<br>auto loss = torch::nn::functional::cross_entropy(logits, targets);

为什么不用 std::log(std::exp(x)) 这类链式调用

因为 std::exp(89) 就已溢出为 inf,而典型 logits 值域常达 ±30~±100。一旦出现 inf,后续 std::log(inf) 是 inf,std::log(0) 是 -inf,loss 失控。

  • log-sum-exp 的平移技巧让所有 exp 输入 ≤ 0,彻底规避溢出
  • 即使全用 double,原始写法在 batch 较大时仍易崩;稳定版可安全跑满 float32 范围
  • 某些嵌入式或纯 C++ 环境无法用 libtorch,这时手写稳定版是唯一可靠选择

真正容易被忽略的是:label 索引越界检查(比如 target >= num_classes)和梯度回传路径——如果只是前向计算 loss,不涉及 autograd,那没问题;但若需反向,必须确保 tensor 在计算图中正确连接。

热门AI工具

更多
WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

讯飞智作

讯飞智作是一款AI视频创作工具,AI文本配音工具,数字人课程、营销视频制作。

音述AI
音述AI Hot

一款AI音频处理工具,主要用于音述AI是一个以“用声音述说故事”为核心的 AI 音乐创作与声音分享社区,适合需要提升相关任务效率的用户。

立刻MV
立刻MV Hot

立刻MV是一款AI文本写作工具,AI 音乐视频(MV)创作工具。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

咔片AIPPT

一款在线AI演示文稿制作工具,可根据主题和内容需求辅助生成PPT结构与页面,提高演示材料制作效率。

Atoms
Atoms Hot

Atoms是一款AI智能体工具,第一支自动构建真实业务的 AI 团队。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

PixPix
PixPix Hot

PixPix是一款面向电商视觉生产的AI商品图生成工具。

相关专题

更多
c++和c语言的区别有哪些
c++和c语言的区别有哪些

c++和c语言的区别:1、面向对象编程(OOP)支持不同;2、新增特性不同;3、标准库不同;4、编译方式不同;5、命名空间不同等等。想了解更多c++的相关内容,可以阅读本专题下面的文章。

2248

2024.03.14

c++和python学习顺序推荐
c++和python学习顺序推荐

一般建议先学习C++,再学习Python,因为这样可以逐步从较为底层的编程语言向更高级的语言过渡。想了解更多python的相关内容,可以阅读本专题下面的文章。

1019

2024.03.14

python和c++学习性价比分析
python和c++学习性价比分析

Python易于学习,广泛应用于Web开发、数据科学和人工智能等领域,但性能较低。C语言性能高,适用于对性能要求较高的场景,如游戏开发和系统编程,但学习曲线陡峭,错误处理复杂。想了解更多python的相关内容,可以阅读本专题下面的文章。

427

2024.03.14

c语言和c++一样吗
c语言和c++一样吗

c语言和c++是两种不同的编程语言,虽然有相似之处,但存在显著差异。c语言专注于过程式编程和系统级开发,以简洁、高效著称。c++作为c语言的超集,引入了面向对象编程,增强了代码组织和管理能力,但学习曲线也更陡峭。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

327

2024.03.14

c语言和c++先学哪个好
c语言和c++先学哪个好

初学者选择学习c语言还是c++语言,需要根据个人学习目标、背景以及编程兴趣和预期应用方向来决定。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

406

2024.03.14

c语言和c++的区别和联系
c语言和c++的区别和联系

c语言和c++是计算机科学领域应用广泛的编程语言。虽然它们有着相似的基础,但它们在语言类型、语法功能和内存管理方面存在着显著差异。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

580

2024.03.14

c++软件中文更改教程
c++软件中文更改教程

对于 ide,可通过打开设置,找到语言设置,选择中文,并保存更改。对于非 ide 应用程序,可查找设置或选项,选择语言设置,更改为中文,并保存更改。想了解更多c++的相关内容,可以阅读本专题下面的文章。

1409

2024.03.21

python和java和c++学习性价比分析
python和java和c++学习性价比分析

Python以其易学性、丰富的库和活跃的社区而著称,适合数据科学、人工智能和Web开发。Java以其跨平台性、企业级应用开发和Android应用开发而闻名。C++以其底层控制能力、高效性能和游戏开发而著称。选择哪种语言取决于个人兴趣、职业方向和特定需求。想了解更多python和java和c++的相关内容,可以阅读本专题下面的文章。

1217

2024.03.22

OpenJDK JVM参数怎么调优
OpenJDK JVM参数怎么调优

Java应用运行缓慢,如何调整JVM参数?本专题介绍OpenJDK JVM性能调优的基本流程,涵盖性能指标确认、运行数据采集、内存配置和垃圾回收分析,帮助你先定位瓶颈,再通过对照测试验证效果,避免直接套用参数而影响应用稳定性。

80

2026.10.16

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
C++ Reference
C++ Reference

共0课时 | 0人学习

C++教程
C++教程

共115课时 | 43.2万人学习

Go语言教程-全程干货无废话
Go语言教程-全程干货无废话

共100课时 | 17.9万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn