Relintai/rcpp_framework
1
0
Fork 0
mirror of https://github.com/Relintai/rcpp_framework.git synced 2024-11-14 04:57:21 +01:00
Code Issues Packages Projects Releases Wiki Activity

Added softmax to Math.

Browse Source
This commit is contained in:
Relintai 2022-03-10 19:20:29 +01:00
parent cc64dee592
commit ed9b23ad86
1 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
core/math/math.h
View File

@ -9,6 +9,8 @@
#include <algorithm> #include <algorithm>
#include <cstdint> #include <cstdint>
#include "core/containers/vector.h"
#define MATH_PI 3.1415926535897932384626433833 #define MATH_PI 3.1415926535897932384626433833
#define EPSILON 0.00001 #define EPSILON 0.00001
@ -343,6 +345,22 @@ public:
static _ALWAYS_INLINE_ double linear_derived(const double x) { static _ALWAYS_INLINE_ double linear_derived(const double x) {
return 1; return 1;
} }
static _ALWAYS_INLINE_ void softmax(Vector<double> *data_in_out) {
double *dwp = data_in_out->dataw();
int s = data_in_out->size();
double acc = 0;
for (int i = 0; i < s; ++i) {
double e = exp(dwp[i]);
dwp[i] = e;
acc += e;
}
for (int i = 0; i < s; ++i) {
dwp[i] /= acc;
}
}
}; };
#ifndef ABS #ifndef ABS