softmax function in c++
#include <iostream> #include <vector> #include <cmath> #include <algorithm> #include <numeric> double myfunction(double num) { return exp(num); } template <typename T> void softmax(const typename::std::vector<T> &v, typename::std::vector<T> &s){ double sum=0.0; transform(v.begin(), v.end(), s.begin(), myfunction); sum=accumulate(s.begin(), s.end(), sum); for(size_t i=0; i<s.size(); ++i) s.at(i)/=sum; } int main() { double a[]={1.0, 3.0, 2.0}; std::vector<double> v_a(a, a+sizeof a/sizeof a[0]), v_b(v_a); std::vector<double>::const_iterator it=v_a.begin(); for(; it!=v_a.end(); ++it) { std::cout<<*it<<" "; } std::cout<<std::endl; softmax(v_a, v_b); it=v_b.begin(); for(; it!=v_b.end(); ++it) { std::cout<<*it<<" "; } std::cout<<std::endl; return 0; }
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· AI与.NET技术实操系列:基于图像分类模型对图像进行分类
· go语言实现终端里的倒计时
· 如何编写易于单元测试的代码
· 10年+ .NET Coder 心语,封装的思维:从隐藏、稳定开始理解其本质意义
· .NET Core 中如何实现缓存的预热?
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· 25岁的心里话
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 按钮权限的设计及实现