macro_rules! argmax {
($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
. See the function documentation for more details.