数组排序后复原

最新推荐文章于 2024-04-26 16:19:48 发布

敲代码的小风

最新推荐文章于 2024-04-26 16:19:48 发布

阅读量703

点赞数

分类专栏：零基础学习SSD网络PyTorch实现 Deep-Learning-with-PyTorch 《深度学习之PyTorch实战计算机视觉》

本文链接：https://blog.csdn.net/m0_46653437/article/details/111699072

版权

零基础学习SSD网络PyTorch实现同时被 3 个专栏收录

293 篇文章 26 订阅

订阅专栏

Deep-Learning-with-PyTorch

216 篇文章 19 订阅

订阅专栏

《深度学习之PyTorch实战计算机视觉》

198 篇文章 41 订阅

订阅专栏

参考链接: pytorch技巧1: 数组排序后复原

Microsoft Windows [版本 10.0.18363.1256]
(c) 2019 Microsoft Corporation。保留所有权利。

C:\Users\chenxuqi>conda activate ssd4pytorch1_2_0

(ssd4pytorch1_2_0) C:\Users\chenxuqi>python
Python 3.7.7 (default, May  6 2020, 11:45:54) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.manual_seed(seed=20200910)
<torch._C.Generator object at 0x00000246CA45D330>
>>>
>>> data = torch.randint(10,(10,))
>>> data
tensor([3, 8, 4, 7, 8, 5, 0, 8, 8, 1])
>>> a, idx1 = torch.sort(data)
>>> a
tensor([0, 1, 3, 4, 5, 7, 8, 8, 8, 8])
>>> idx1
tensor([6, 9, 0, 2, 5, 3, 1, 4, 7, 8])
>>>
>>> data
tensor([3, 8, 4, 7, 8, 5, 0, 8, 8, 1])
>>> b, idx2 = torch.sort(idx1)
>>> b
tensor([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
>>> idx2
tensor([2, 6, 3, 5, 7, 4, 0, 8, 9, 1])
>>>
>>>
>>> a.index_select(0,idx2)
tensor([3, 8, 4, 7, 8, 5, 0, 8, 8, 1])
>>>
>>>
>>>