共计 385 个字符,预计需要花费 1 分钟才能阅读完成。
2D 最大池化
代码
import torch
import torch.nn as nn
input = torch.randn(1, 3, 224, 224)
maxpooling2d = nn.MaxPool2d(3, 2)
output = maxpooling2d(input)
print(output.size())
torch.Size([1, 3, 111, 111])
import torch
import torch.nn as nn
input = 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+cu101
torchvision=0.8.2
torchaudio=0.7.2
正文完