Function argmax 

pub fn argmax(a: &Array, axis: Option<isize>, keepdims: bool) -> Result<Array>

Index (uint32) of the maximum along axis, or over the flattened array if None. The first occurrence wins; the first NaN wins over any number.