See https://pytorch.org/docs/stable/generated/torch.nn.AvgPool1d.html#avgpool2d and https://pytorch.org/docs/stable/generated/torch.nn.AdaptiveAvgPool2d.html#torch.nn.AdaptiveAvgPool2d for reference.