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