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) 返回类型: 数组