jax.numpy.msort

jax.numpy.msort(a)

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

See also

sort()

Notes

np.msort(a) is equivalent to np.sort(a, axis=0).