megengine.functional.argsort

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

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

参数

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

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

  • inp – Tensor:

  • descending – bool:

返回类型

Tensor

返回

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

实际案例

>>> import numpy as np
>>> x = Tensor(np.array([1,2], dtype=np.float32))
>>> F.argsort(x)
Tensor([0 1], dtype=int32, device=xpux:0)

上一页

megengine.functional.sort

下一页

megengine.functional.topk