msort

paddle. msort ( input: Tensor, *, out: Tensor | None = None ) [源代码] 

沿输入 Tensor 的第 0 轴（axis=0）按升序对元素进行排序。

该函数等价于 paddle.sort(x, axis=0)。

参数

input (Tensor) - 输入的 N-D Tensor，支持的数据类型为：float32、float64、int16、int32、int64、uint8。别名 input。

name (str，可选) - 具体用法请参见 api_guide_Name，一般无需设置，默认值为 None。

关键字参数

out (Tensor，可选) - 输出 Tensor，若不为 None，计算结果将保存在该 Tensor 中，默认值为 None。

返回

Tensor，排序后的输出（与 input 维度相同、数据类型相同）。

代码示例

          >>> import paddle

>>> x = paddle.to_tensor([[[5,8,9,5],
...                        [0,0,1,7],
...                        [6,9,2,4]],
...                       [[5,2,4,2],
...                        [4,7,7,9],
...                        [1,7,0,6]]],
...                      dtype='float32')
>>> out1 = paddle.msort(input=x)
>>> print(out1.numpy())
[[[5. 2. 4. 2.]
  [0. 0. 1. 7.]
  [1. 7. 0. 4.]]
 [[5. 8. 9. 5.]
  [4. 7. 7. 9.]
  [6. 9. 2. 6.]]]

>>> out2 = paddle.empty_like(x)
>>> paddle.msort(input=x, out=out2)
>>> print(out2.numpy())
[[[5. 2. 4. 2.]
  [0. 0. 1. 7.]
  [1. 7. 0. 4.]]
 [[5. 8. 9. 5.]
  [4. 7. 7. 9.]
  [6. 9. 2. 6.]]]

         