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

◆ nn_softmax1d_f32()

void nn_softmax1d_f32 ( Tensor1D_F32 y,
const Tensor1D_F32 x 
)

Applies the softmax activation function to a 1D floating-point tensor.

nn_softmax1d_f32

y[i] = exp(x[i]) / sum(exp(x))

Parameters
yThe result tensor.
xThe input tensor.