arrayfire

Function imax

Source 
pub fn imax<T>(input: &Array<T>, dim: i32) -> (Array<T::InType>, Array<u32>)
where
    T: HasAfEnum,
    T::InType: HasAfEnum,
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.