Python的torch.sort()如何使用

這篇文章主要介紹“Python的torch.sort()如何使用”，在日常操作中，相信很多人在Python的torch.sort()如何使用問題上存在疑惑，小編查閱了各式資料，整理出簡單好用的操作方法，希望對大家解答”Python的torch.sort()如何使用”的疑惑有所幫助！接下來，請跟著小編一起來學習吧！

代碼實驗展示:

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

C:\Users\chenxuqi>conda activate ssd4pytorch2_2_0(ssd4pytorch2_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 0x00000144A347D330>>>>>>>>>> x = torch.randn(3, 4)>>> x
tensor([[ 0.2824, -0.3715,  0.9088, -1.7601],[-0.1806,  2.0937,  1.0406, -1.7651],[ 1.1216,  0.8440,  0.1783,  0.6859]])>>>>>> sorted, indices = torch.sort(x)>>> sortedtensor([[-1.7601, -0.3715,  0.2824,  0.9088],[-1.7651, -0.1806,  1.0406,  2.0937],[ 0.1783,  0.6859,  0.8440,  1.1216]])>>> indices
tensor([[3, 1, 0, 2],[3, 0, 2, 1],[2, 3, 1, 0]])>>>>>>>>>>>>>>> sorted, indices = torch.sort(x, 0)>>> sortedtensor([[-0.1806, -0.3715,  0.1783, -1.7651],[ 0.2824,  0.8440,  0.9088, -1.7601],[ 1.1216,  2.0937,  1.0406,  0.6859]])>>> indices
tensor([[1, 0, 2, 1],[0, 2, 0, 0],[2, 1, 1, 2]])>>> x
tensor([[ 0.2824, -0.3715,  0.9088, -1.7601],[-0.1806,  2.0937,  1.0406, -1.7651],[ 1.1216,  0.8440,  0.1783,  0.6859]])>>>>>>>>>

到此，關于“Python的torch.sort()如何使用”的學習就結束了，希望能夠解決大家的疑惑。理論與實踐的搭配能更好的幫助大家學習，快去試試吧！若想繼續學習更多相關知識，請繼續關注億速云網站，小編會繼續努力為大家帶來更多實用的文章！