mlx_rs::ops

Function softmax_all

Source
pub fn softmax_all(
    a: impl AsRef<Array>,
    precise: impl Into<Option<bool>>,
) -> Result<Array>
Expand description

Perform the softmax along all axes.