fun <T : Any> KtNDArray<T>.argPartition(kth: IntArray, axis: Int? = -1, kind: String = "introselect"): KtNDArray<Long> (source)
Returns the indices that would partition this array.
kth - Element index to partition by.
The k-th element will be in its final sorted position and all smaller elements will be moved before it and
all larger elements behind it. The order all elements in the partitions is undefined.
If provided with a sequence of k-th it will partition all of them into their sorted position at once.
axis - Axis along which to sort. The default is -1 (the last axis). If null, the flattened array is used.
kind - Selection algorithm. Default is ‘introselect’.
Return
KtNDArray Array of indices that partition this array along the specified axis.