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;
}
复制代码

 

posted @   东宫得臣  阅读(1728)  评论(0编辑  收藏  举报
编辑推荐:
· AI与.NET技术实操系列:基于图像分类模型对图像进行分类
· go语言实现终端里的倒计时
· 如何编写易于单元测试的代码
· 10年+ .NET Coder 心语,封装的思维:从隐藏、稳定开始理解其本质意义
· .NET Core 中如何实现缓存的预热?
阅读排行:
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· 25岁的心里话
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 按钮权限的设计及实现
点击右上角即可分享
微信分享提示