softmax函数

Posted 2021-02-17 smallredness

#include <vector>
#include <cmath> //math.h

std::vector<double> vsoftmax(std::vector<double> &v) {
    double sum=0;
    for(auto iter:v) {
        sum+=exp(iter);
    }

    std::vector<double> res;
    for(int i=0;i<v.size();i++) {
        res.push_back(exp(v[i])/sum);
    }

    return res;
}