2D最大池化
代码
import torchimport torch.nn as nninput = torch.randn(1, 3, 224, 224)maxpooling2d = nn.MaxPool2d(3, 2)output = maxpooling2d(input)print(output.size())torch.Size([1, 3, 111, 111])import torchimport torch.nn as nninput = torch.randn(1, 3, 224, 224)maxpooling2d = nn.MaxPool2d(3, 2, 1)output = maxpooling2d(input)print(output.size())torch.Size([1, 3, 112, 112])援用
<1>
torch=1.7.1+cu101torchvision=0.8.2torchaudio=0.7.2