|
Baremetal-NN
Baremetal-NN API documentation
|
| void nn_elu2d_f32 | ( | Tensor2D_F32 * | y, |
| const Tensor2D_F32 * | x, | ||
| float | alpha | ||
| ) |
Applies the ELU activation function to a 2D floating-point tensor.
nn_elu2d_f32
y[i][j] = x[i][j] if x[i][j] > 0 else alpha * (exp(x[i][j]) - 1)
| y | The result tensor. |
| x | The input tensor. |
| alpha | The alpha parameter. |