Function imax
Source pub fn imax<T>(input: &Array<T>, dim: i32) -> (Array<T::InType>, Array<u32>)
Expand description
Find maximum value along given dimension and their corresponding indices
§Parameters
input - Input Array
dim - Dimension along which the input Array will be reduced
§Return Values
A tuple of Arrays: Array maximum values and Array containing their index along the reduced dimension.