jax.Array.argpartition

abstract Array.argpartition(kth, axis=-1)

傳回部分排序陣列的索引。

請參閱 jax.numpy.argpartition() 以取得完整文件。

參數:

• self (Array)

• kth (int)

• axis (int)

傳回類型:

Array