jax.numpy.msort#
- jax.numpy.msort(a)[source]#
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