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

◆ NN_softmax()

void NN_softmax ( Tensor out,
const Tensor tensor,
int  dim 
)

Returns the sum of all elements in the input tensor.

Parameters
outthe output scalar tensor
tensorthe input tensor
dimthe dimension to reduce