megengine.functional.argsort¶

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

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

参数

inp (Tensor) – 输入张量。如果是2维，返回结果为如何为输入张量按行排序的一组下标。
descending (bool) – 降序排列，即最大值位于第一位置。默认：False
inp – Tensor:
descending – bool:

返回类型

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