mindspore.ops.msort
- mindspore.ops.msort(input)[源代码]
将输入Tensor的元素沿其第一个维度按值升序排序。
ops.msort(t)相当于ops.Sort(axis=0)(t)[0]。另外可以参考
mindspore.ops.Sort()
。- 参数:
input (Tensor) - 需要排序的输入,类型必须是float16或者float32。
- 返回:
排序后的Tensor,与输入的shape和dtype一致。
- 异常:
TypeError - input 的类型既不是float16也不是float32。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore as ms >>> from mindspore import ops >>> import numpy as np >>> input = ms.Tensor(np.array([[8, 2, 1], [5, 9, 3], [4, 6, 7]]), ms.float16) >>> output = ops.msort(input) >>> print(output) [[4. 2. 1.] [5. 6. 3.] [8. 9. 7.]]