jax.Array.argsort

abstract Array.argsort(axis=-1, *, kind=None, order=None, stable=True, descending=False)

傳回排序陣列的索引。

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

參數:

• self (Array)

• axis (int | None)

• kind (None)

• order (None)

• stable (bool)

• descending (bool)

傳回類型:

Array