Skip to main content

argmax

ferray

Function argmax 

Source 
pub fn argmax<T, D>(
    a: &Array<T, D>,
    axis: Option<usize>,
) -> Result<Array<u64, IxDyn>, FerrayError>
where
    T: Element + PartialOrd + Copy,
    D: Dimension,
Expand description

Index of the maximum value.

Equivalent to numpy.argmax.