Function argmax 

pub fn argmax(a: &Array, axis: impl Axis) -> Array

Indices (uint32) of the maxima along one axis (..: the flattened array); see Array::argmax.