megengine.functional.sort

sort(inp, descending=False)[源代码]

返回经过排序后的张量。

参数
  • inp – 输入张量,如果是二维的,则该张量的每行将进行排序。

  • descending – 降序排列,即最大值位于第一位置。 默认:False

返回

由两个张量组成的元组 (sorted_tensor, indices_of_int32)

例如:

import numpy as np
from megengine import tensor
import megengine.functional as F

x = tensor(np.array([1,2], dtype=np.float32))
out, indices = F.sort(x)
print(out.numpy())

输出:

[1. 2.]