Function argpartition 

pub fn argpartition(a: &Array, kth: i32, axis: impl Axis) -> Array

The indices (uint32) that partition the array; see Array::argpartition.