python torch.max() 用法记录

目录

一、试验记录

二、功能

---

相遇的地点：这个函数很好用，在NMS中遇到了它。ο(=•ω＜=)ρ⌒☆ 

我一路向北，离开有你的季节~~*★,°*:.☆(￣▽￣)/$:*.°★* 。

一、试验记录

举例如下

1、

import torch

a = torch.tensor([[1,2,3],[1,3,5],[1,2,4],[0,7,2]])
print(a)
print(a.shape)
print('======')
a1, a2 = a.max(1, keepdim=True)
print(a1)
print(a2)

输出为

tensor([[1, 2, 3],
        [1, 3, 5],
        [1, 2, 4],
        [0, 7, 2]])
torch.Size([4, 3])
======
tensor([[3],
        [5],
        [4],
        [7]])
tensor([[2],
        [2],
        [2],
        [1]])

2、 

import torch

a = torch.tensor([[[1,2,3],[1,3,5],[1,2,4],[0,7,2]],[[2,8,6],[5,0,8],[1,2,4],[8,5,2]]])
print(a)
print(a.shape)
print('======')
a1, a2 = a.max(1, keepdim=True)
print(a1)
print(a2)

输出

tensor([[[1, 2, 3],
         [1, 3, 5],
         [1, 2, 4],
         [0, 7, 2]],

        [[2, 8, 6],
         [5, 0, 8],
         [1, 2, 4],
         [8, 5, 2]]])
torch.Size([2, 4, 3])
======
tensor([[[1, 7, 5]],

        [[8, 8, 8]]])
tensor([[[0, 3, 1]],

        [[3, 0, 1]]])

二、功能

返回 dim维度 中 最大的值，以及在该 维度 中的位置。