org.apache.mxnet.javaapi
This Param Object is specifically used for softmax
The input array.
The axis along which to compute softmax.
DType of the output in case this can't be inferred. Defaults to the same as input's dtype if not defined (dtype=None).
Temperature parameter in softmax
This Param Object is specifically used for softmax