jax.Array.argpartition# abstract Array.argpartition(kth, axis=-1)[源代码]# 返回部分排序数组的索引。 请参阅 jax.numpy.argpartition() 以获取完整文档。 参数: self (Array) kth (int) axis (int) 返回类型: Array
