Function argmax_keepdims 

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

argmax keeping the reduced axis; see Array::argmax_keepdims.