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
See also
Notes
np.msort(a)
is equivalent tonp.sort(a, axis=0)
.