torch.argsort

torch.argsort(input, dim=- 1, descending=False, stable=False) → Tensor

返回沿给定维度按值升序排序张量的索引。

这是该方法的第二个返回值。请参阅其文档以了解该方法的精确语义。

如果 stable 是 True ，则排序例程将变得稳定，保留等效元素的顺序。如果 False ，则比较相等的值的相对顺序不保证。 True 较慢。

参数:

• input (Tensor) – 输入张量。

• dim（int，可选）- 要排序的维度

• descending（bool，可选）- 控制排序顺序（升序或降序）

• 稳定（布尔值，可选）- 控制等效元素的相对顺序

示例：

>>> a = torch.randn(4, 4)
>>> a
tensor([[ 0.0785,  1.5267, -0.8521,  0.4065],
        [ 0.1598,  0.0788, -0.0745, -1.2700],
        [ 1.2208,  1.0722, -0.7064,  1.2564],
        [ 0.0669, -0.2318, -0.8229, -0.9280]])


>>> torch.argsort(a, dim=1)
tensor([[2, 0, 3, 1],
        [3, 2, 1, 0],
        [2, 1, 0, 3],
        [3, 2, 1, 0]])