cupy.argpartition#

cupy.argpartition(a, kth, axis=-1)[source]#

返回对数组进行部分排序的索引。

参数:

  • a (cupy.ndarray) – 要排序的数组。

  • kth (int 或 int 序列) – 用于分区的元素索引。如果提供一个 k-th 序列，它将一次性将所有由这些 k-th 索引的元素分区到其排序位置。

  • axis (int 或 None) – 排序轴。默认值为 -1，表示沿最后一个轴排序。如果提供 None，则在排序前会将数组展平。

返回值:

类型和形状与 a 相同的数组。

返回类型:

cupy.ndarray

注意

出于实现原因，cupy.argpartition 会像 cupy.argsort 一样对给定数组进行完全排序。它也不支持 numpy.argpartition 支持的 kind 和 order 参数。

另请参阅

numpy.argpartition()