[函数]nn.AdaptiveAvgPool2d

class torch.nn.AdaptiveAvgPool2d(output_size)

对由多个输入平面组成的输入信号应用二维自适应平均池 (2D adaptive average pooling )。

对于任何输入大小,输出的大小都是h\times w

输出特征图的数量等于输入特征图的数量。

猜你喜欢

转载自blog.csdn.net/weixin_40519315/article/details/105323144