torch.argmin()的使用举例

参考链接: argmin(dim=None, keepdim=False) → LongTensor
参考链接: torch.argmin()

代码实验举例:

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 0x000001FE6BDED330>
>>>
>>> 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])
>>>
>>>
>>>