Function softmax

pub fn softmax(
    a: impl AsRef<Array>,
    axes: &[i32],
    precise: impl Into<Option<bool>>,
) -> Result<Array>

Perform the softmax along the given axis.

See the python API docs for more information.