jax.Array.sort#

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

Return a sorted copy of an array.

Refer tojax.numpy.sort() for full documentation.

Parameters:

• self (Array)

• axis (int |None)

• kind (None)

• order (None)

• stable (bool)

• descending (bool)

Return type:

Array