torch.argmin()怎么使用

本篇内容主要讲解“torch.argmin()怎么使用”，感兴趣的朋友不妨来看看。本文介绍的方法操作简单快捷，实用性强。下面就让小编来带大家学习“torch.argmin()怎么使用”吧!

创新互联专业为企业提供青河网站建设、青河做网站、青河网站设计、青河网站制作等企业网站建设、网页设计与制作、青河企业网站模板建站服务，十多年青河做网站经验，不只是建网站，更提供有价值的思路和整体网络服务。

代码实验举例:

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, 5)>>> 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]])>>> torch.argmin(a)tensor(17)>>> torch.argmin(a, dim=1)tensor([3, 1, 3, 2])>>> torch.argmin(a, dim=0)tensor([3, 1, 3, 2, 2])>>>>>>>>>

到此，相信大家对“torch.argmin()怎么使用”有了更深的了解，不妨来实际操作一番吧！这里是创新互联网站，更多相关内容可以进入相关频道进行查询，关注我们，继续学习！