megengine.functional.sort¶

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

返回经过排序后的张量，以及对应原始张量中的索引位置。

参数

inp (Tensor) – 输入张量，如果是二维的，则该张量的每行将进行排序。
descending (bool) – 降序排列，即最大值位于第一位置。默认：False

返回类型

Tuple[Tensor, Tensor]

返回

由两个张量组成的元组 (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.]

megengine.functional.argmax

megengine.functional.argsort