@@ -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;
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user