jax.Array.sort

Contents

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