jax.Array.sort — JAX documentation (original) (raw)

jax.Array.sort#

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

Return a sorted copy of an array.

Refer to jax.numpy.sort() for full documentation.

Parameters:

• self (Array)
• axis (int | None)
• kind (None)
• order (None)
• stable (bool)
• descending (bool)

Return type:

Array