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
out
the output scalar tensor
tensor
the input tensor
dim
the dimension to reduce
nn
functional
softmax.h
Generated by
1.9.5