mindspore.ops.msort

mindspore.ops.msort(input)[源代码]

将输入Tensor的元素沿其第一个维度按值升序排序。

ops.msort(t)相当于ops.Sort(axis=0)(t)[0]。另外可以参考 mindspore.ops.Sort()。

说明

当前Ascend后端只支持对一维输入进行排序。

参数：

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.]]