jax.Array.sort

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

返回数组的排序副本。

请参阅 jax.numpy.sort() 获取完整文档。

参数:

• self (Array)

• axis (int | None)

• kind (None)

• order (None)

• stable (bool)

• descending (bool)

返回类型:

Array