megengine.functional.argsort

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

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

参数
  • inp – 输入张量。如果是2维,返回结果为如何为输入张量按行排序的一组下标。

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

返回

类型为 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]