megengine.functional.argsort

argsort(inp, descending=False)

返回一个与输入张量形状相同的全1张量。

参数

• inp (Tensor) – 输入张量。如果是2维，返回结果为如何为输入张量按行排序的一组下标。

• descending (bool) – 降序排列，即最大值位于第一位置。 默认：False

返回类型

Tensor

返回

类型为 int32 的下标，展示如何为输入排序。

例如：

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

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

输出：

[0 1]

megengine.functional.sort megengine.functional.topk