gpt4 book ai didi

conv-neural-network - 如何在 PyTorch 中执行 sum pooling

转载 作者:行者123 更新时间:2023-12-05 00:40:18 28 4
gpt4 key购买 nike

如何在 PyTorch 中执行 sum pooling。具体来说,如果我们输入 (N, C, W_in, H_in) 并希望使用特定的 kernel_size 输出 (N, C, W_out, H_out) code> 和 stride 就像 nn.Maxpool2d ?

最佳答案

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

关于conv-neural-network - 如何在 PyTorch 中执行 sum pooling,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/50838876/

28 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com