argsort

paddle.argsort ( x, axis=- 1, descending=False, name=None ) [源代码]

对输入变量沿给定轴进行排序，输出排序好的数据的相应索引，其维度和输入相同。默认升序排列，如果需要降序排列设置 descending=True 。

参数：

  • x (Tensor) - 输入的多维 Tensor ，支持的数据类型：float32、float64、int16、int32、int64、uint8。

  • axis (int，可选) - 指定对输入Tensor进行运算的轴， axis 的有效范围是[-R, R)，R是输入 x 的Rank， axis 为负时与 axis +R 等价。默认值为0。

  • descending (bool，可选) - 指定算法排序的方向。如果设置为True，算法按照降序排序。如果设置为False或者不设置，按照升序排序。默认值为False。

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

返回：Tensor, 排序后索引信息（与 x 维度信息一致），数据类型为int64。

代码示例：

  1. import paddle
  3. x = paddle.to_tensor([[[5,8,9,5],
  4.                   [0,0,1,7],
  5.                   [6,9,2,4]],
  6.                   [[5,2,4,2],
  7.                   [4,7,7,9],
  8.                   [1,7,0,6]]], dtype='float32')
  9. out1 = paddle.argsort(x=x, axis=-1)
  10. out2 = paddle.argsort(x=x, axis=0)
  11. out3 = paddle.argsort(x=x, axis=1)
  12. print(out1)
  13. #[[[0 3 1 2]
  14. #  [0 1 2 3]
  15. #  [2 3 0 1]]
  16. # [[1 3 2 0]
  17. #  [0 1 2 3]
  18. #  [2 0 3 1]]]
  19. print(out2)
  20. #[[[0 1 1 1]
  21. #  [0 0 0 0]
  22. #  [1 1 1 0]]
  23. # [[1 0 0 0]
  24. #  [1 1 1 1]
  25. #  [0 0 0 1]]]
  26. print(out3)
  27. #[[[1 1 1 2]
  28. #  [0 0 2 0]
  29. #  [2 2 0 1]]
  30. # [[2 0 2 0]
  31. #  [1 1 0 2]
  32. #  [0 2 1 1]]]