本篇内容主要讲解“Python torch.argmax()怎么使用”，感兴趣的朋友不妨来看看。本文介绍的方法操作简单快捷，实用性强。下面就让小编来带大家学习“Python torch.argmax()怎么使用”吧!
创新互联公司2013年成立，先为中方等服务建站，中方等地企业，进行企业商务咨询服务。为中方企业网站制作PC+手机+微官网三网同步一站式服务解决您的所有建站问题。
代码实验展示:
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)>>>>>> 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)>>>>>> 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()怎么使用”有了更深的了解，不妨来实际操作一番吧！这里是创新互联网站，更多相关内容可以进入相关频道进行查询，关注我们，继续学习！
当前名称：Pythontorch.argmax()怎么使用
本文地址：
http://jkwzsj.com/article/iejhsc.html
