jax.numpy.msort
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.
- Parameters
a (array_like) – Array to be sorted.
- Returns
sorted_array – Array of the same type and shape as a.
- Return type