mlx_rs

Macro take_along_axis

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

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