autograd::ops
pub fn softmax(x: &Tensor, axis: isize) -> Tensor
Softmax function.
Take softmax along axis.
axis