pub fn softmax_device(
a: impl AsRef<Array>,
axes: &[i32],
precise: impl Into<Option<bool>>,
stream: impl AsRef<Stream>,
) -> Result<Array>Expand description
Perform the softmax along the given axis.
See the python API docs for more information.