【pytorch 中 torch.max 和 torch.argmax 的区别】

torch.max 和 torch.argmax 的区别

1.torch.max

torch.max(input, dim, max=None, max_indices=None, keepdim=False) -->> (Tensor, LongTensor)

作用：找出给定tensor的指定维度dim上的上的最大值，并返回最大值在该维度上的值和位置索引。
应用举例
例1——返回相应维度上的最大值,并返回最大值的位置索引

a = torch.randn(4, 4)
a
>tensor([[-1.2360, -0.2942, -0.1222,  0.8475],
            [ 1.1949, -1.1127, -2.2379, -0.6702],
            [ 1.5717, -0.9207,  0.1297, -1.8768],
            [-0.6172,  1.0036, -0.6060, -0.2432]])
torch.max(a, 1)
>torch.return_types.max(values=tensor([0.8475, 1.1949, 1.5717, 1.0036]), 
indices=tensor([3, 0, 0, 1]))
1
2
3
4
5
6
7
8
9

例2——如果max的参数只有一个tensor，则返回该tensor里所有值中的最大值。

a = torch.randn(4, 4)
a
>tensor([[ 0.4997,  0.8054,  0.1761,  0.3055],
        [-1.2234,  0.3823,  0.2266, -2.9062],
        [ 0.4390, -1.0142, -0.5314, -1.7095],
        [-0.2296, -0.4230, -0.7446, -0.0828]])
torch.max(a)
>tensor(0.8054)

1
2
3
4
5
6
7
8
9

例3——如果max的参数是两个相同shape的tensor，则返回两tensor元素对应位置的最大值的新tensor

a = torch.randint(2, 10,(6,4))
a
>tensor([[8, 7, 3, 5],
        [2, 8, 3, 4],
        [3, 2, 5, 5],
        [4, 7, 5, 2],
        [2, 9, 3, 8],
        [4, 4, 2, 2]])
b = torch.randint(2, 10,(6,4))
b
>tensor([[9, 8, 9, 2],
        [4, 3, 3, 4],
        [6, 9, 2, 7],
        [4, 3, 2, 7],
        [4, 4, 9, 2],
        [8, 2, 6, 2]])
torch.max(a, b) 
>tensor([[9, 8, 9, 5],
        [4, 8, 3, 4],
        [6, 9, 5, 7],
        [4, 7, 5, 7],
        [4, 9, 9, 8],
        [8, 4, 6, 2]])
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

2. torch.argmax

函数定义
torch.argmax(input, dim, keepdim=False) → LongTensor
作用：返回输入张量中指定维度的最大值的索引。
应用举例：
例1——指定维度：返回相应维度最大值的索引

a = torch.randn(4, 4)
a
>tensor([[ 1.3398,  0.2663, -0.2686,  0.2450],
            [-0.7401, -0.8805, -0.3402, -1.1936],
            [ 0.4907, -1.3948, -1.0691, -0.3132],
            [-1.6092,  0.5419, -0.2993,  0.3195]])
torch.argmax(a, dim=1)
>tensor([ 0,  2,  0,  1])
1
2
3
4
5
6
7
8

例2——不指定维度，返回整体上最大值的序号

a = torch.randint(9,(3, 3))
a
>tensor([[5, 2, 2],
         [7, 2, 0],
         [8, 0, 6]])
torch.argmax(a)
>tensor(6)
1
2
3
4
5
6
7

3.torch.min

用法同max

4.torch.argmin

用法同argmax