pub fn partition(a: &Array, kth: isize, axis: Option<isize>) -> Result<Array>Expand description
a partitioned along axis (flattened without one): the element at position kth
is the one a sort would put there, smaller ones come before it and larger ones after
(MLX’s partition).