jax.numpy.msort

jax.numpy.msort(a)

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

LAX-backend implementation of numpy.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