Python torch.argmax()怎么使用

本篇內容主要講解“Python torch.argmax()怎么使用”，感興趣的朋友不妨來看看。本文介紹的方法操作簡單快捷，實用性強。下面就讓小編來帶大家學習“Python torch.argmax()怎么使用”吧!

代碼實驗展示:

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 0x000002019B2DD330>>>>>>> a = torch.randn(4, 4)>>> a
tensor([[ 0.5816,  2.0060,  1.6013, -0.6379],[-1.1943,  0.1426,  1.3612, -1.4171],[-2.7679,  2.1961,  1.3462, -0.7650],[ 1.5528,  0.7210, -0.1314,  0.1448]])>>> torch.argmax(a)tensor(9)>>>>>> torch.manual_seed(seed=20200910)<torch._C.Generator object at 0x000002019B2DD330>>>>>>> a = torch.randn(4, 5)>>> torch.argmax(a, dim=1)tensor([1, 3, 1, 4])>>> torch.argmax(a, dim=0)tensor([2, 2, 0, 1, 3])>>> a
tensor([[ 0.5816,  2.0060,  1.6013, -0.6379, -0.4996],[ 0.2791, -0.5900,  0.2527,  0.2797, -0.5316],[ 0.6480,  2.6538, -0.2797, -1.3232, -0.8263],[-0.4343, -0.2370, -1.7016, -0.2565,  1.4568]])>>>>>>>>>

到此，相信大家對“Python torch.argmax()怎么使用”有了更深的了解，不妨來實際操作一番吧！這里是億速云網站，更多相關內容可以進入相關頻道進行查詢，關注我們，繼續學習！