Warm tip: This article is reproduced from serverfault.com, please click

conv neural network-如何在PyTorch中执行总池

(conv neural network - How to perform sum pooling in PyTorch)

发布于 2018-06-13 13:46:51

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

Questioner
adeelz92
Viewed
0
benjaminplanche 2018-06-13 22:20:43

你可以使用torch.nn.AvgPool1d(或torch.nn.AvgPool2dtorch.nn.AvgPool3d)执行均值合并-与总合并成比例。如果确实需要求和的值,则可以将平均输出乘以池化表面。