mlx_rs

Macro argmax_axis

Source 
macro_rules! argmax_axis {
    ($a:expr, $axis:expr) => { ... };
    ($a:expr, $axis:expr, stream = $stream:expr) => { ... };
    ($a:expr, $axis:expr, keep_dims = $keep_dims:expr) => { ... };
    ($a:expr, $axis:expr, keep_dims = $keep_dims:expr, stream = $stream:expr) => { ... };
}
Expand description

Macro generated for the function crate::ops::indexing::argmax_axis. See the function documentation for more details.