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.

Deprecated since version 1.24: msort is deprecated, use np.sort(a, axis=0) instead.

Parameters:

a (array_like) – Array to be sorted.

Returns:

sorted_array – Array of the same type and shape as a.

Return type:

ndarray