torch.sort()和torch.argsort()函数解析

BILLY BILLY

于 2024-09-16 13:05:32 发布

阅读量211

点赞数 10

分类专栏： torch基础文章标签： pytorch python

本文链接：https://blog.csdn.net/m0_51579041/article/details/142301630

版权

torch基础专栏收录该内容

13 篇文章 0 订阅

订阅专栏

一. torch.sort()函数解析

1. 官网解释

在这里插入图片描述

2. torch.sort()函数解析

torch.sort(input, dim=- 1, descending=False, stable=False, *, out=None)

输入input，在dim维进行排序，默认是dim=-1对最后一维进行排序，descending表示是否按降序排,默认为False，输出排序后的值以及对应值在原输入imput中的下标

3. 代码举例

dim = -1 表示对每行中的元素进行升序排序，descending=False表示升序排序

x = torch.randn(3, 4)
sorted, indices = torch.sort(x)
x,sorted,indices

输出结果如下：

(tensor([[-1.3864,  0.5811, -0.1056, -0.3237],
         [-0.2136, -1.4806,  0.4986,  0.9382],
         [-0.2820,  0.1171, -0.3983, -0.8061]]),
 tensor([[-1.3864, -0.3237, -0.1056,  0.5811],
         [-1.4806, -0.2136,  0.4986,  0.9382],
         [-0.8061, -0.3983, -0.2820,  0.1171]]),
 tensor([[0, 3, 2, 1],
         [1, 0, 2, 3],
         [3, 2, 0, 1]]))

二.torch.argsort()函数解析

1. 官网解释

torch.argsort()，如下图所示：
在这里插入图片描述

2. torch.argsort()函数解析

用法跟上面torch.sort()函数一样，不同的是torch.argsort()返回只是排序后的值所对应原输入input的下标，即torch.sort()返回的indices

3. 代码举例

dim = 1 表示对每行中的元素进行降序排序，descending=True表示降序排序，输出结果为返回排序后的值所对应原输入input的下标indices

x = torch.randn(3, 4)
indices = torch.argsort(x,dim=1,descending=True)
x,indices

输出结果如下：

(tensor([[-0.6069, -0.9252, -0.9177,  0.6997],
         [ 0.3245, -0.0665,  0.4600,  0.0722],
         [-1.0662,  2.2669, -0.1171, -0.9208]]),
 tensor([[3, 0, 2, 1],
         [2, 0, 3, 1],
         [1, 2, 3, 0]]))