simplify softmax

This commit is contained in:
Marco Cognetta
2023-07-24 10:51:46 +09:00
parent 7d401d530c
commit 80679e24be
+3 -4
View File
@@ -207,15 +207,14 @@ void softmax(float* x, int size) {
max_val = x[i]; max_val = x[i];
} }
} }
// e^x
for (int i = 0; i < size; i++) {
x[i] = exp(x[i] - max_val);
}
// normalize // normalize
float sum = 0.0f; float sum = 0.0f;
for (int i = 0; i < size; i++) { for (int i = 0; i < size; i++) {
x[i] = exp(x[i] - max_val);
sum += x[i]; sum += x[i];
} }
for (int i = 0; i < size; i++) { for (int i = 0; i < size; i++) {
x[i] /= sum; x[i] /= sum;
} }