AdaptiveMaxPool
AdaptiveMaxPool是PyTorch中提供的自适应池化层。
其主要特殊的地方在于:
无论输入Input
的size是多少,输出的size总为指定的size。
AdaptiveMaxPool1d()
m = nn.AdaptiveMaxPool1d(3)
input = torch.randn(4,3,7)
output = m(input) # output的size为(4,3,3)
AdaptiveMaxPool2d()
m = nn.AdaptiveMaxPool2d((3,6))
input = torch.randn(2,64,8,9)
output = m(input) # output的size为(2,64,3,6)
AdaptiveMaxPool3d()
同理