如何在 PyTorch 中执行求和池

Question

如何在 PyTorch 中执行求和池。具体来说，如果我们有输入(N, C, W_in, H_in)并想要(N, C, W_out, H_out)使用特定的输出kernel_size，stride就像nn.Maxpool2d?

Answer 1

您可以使用torch.nn.AvgPool1d(or torch.nn.AvgPool2d, torch.nn.AvgPool3d) 执行平均池化 - 与总和池化成正比。如果您真的想要求和值，您可以将平均输出乘以池化表面。