Как выполнить объединение сумм в 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 (или torch.nn.AvgPool2d, torch.nn.AvgPool3d), которые выполняют среднее объединение - пропорционально суммированию.Если вам действительно нужны суммированные значения, вы можете умножить усредненный результат на поверхность пула.