jax.Array.argsort

jax.Array.argsort#

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

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

有关完整文档,请参阅 jax.numpy.argsort()

参数:
  • self (Array)

  • axis (int | None)

  • kind (None)

  • order (None)

  • stable (bool)

  • descending (bool)

返回类型:

数组