Baremetal-NN
Baremetal-NN API documentation
Loading...
Searching...
No Matches

◆ nn_elu2d_f32()

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)

Parameters
yThe result tensor.
xThe input tensor.
alphaThe alpha parameter.