模型开发接口
推理部署接口
API 相关说明
返回经过排序后的张量,以及对应原始张量中的索引位置。
inp (Tensor) – 输入张量,如果是二维的,则该张量的每行将进行排序。
Tensor
descending (bool) – 降序排列,即最大值位于第一位置。 默认:False
bool
Tuple[Tensor, Tensor]
Tuple
由两个张量组成的元组 (sorted_tensor, indices_of_int32)。
实际案例
>>> import numpy as np >>> x = Tensor(np.array([1,2], dtype=np.float32)) >>> out, indices = F.sort(x) >>> out.numpy() array([1., 2.], dtype=float32)
上一页
megengine.functional.argmax
下一页
megengine.functional.argsort