jax.numpy.msort

Warning

This page was created from a pull request (#9655).

jax.numpy.msort¶

jax.numpy.msort(a)[source]¶

Return a copy of an array sorted along the first axis.

LAX-backend implementation of msort().

Original docstring below.

Parameters: a (array_like) – Array to be sorted.
Returns: sorted_array – Array of the same type and shape as a.
Return type: ndarray

previous

jax.numpy.moveaxis

next

jax.numpy.multiply