PyTorch AdaptiveAvgPool()

对输入,提供自适应平均池化操作。

对于任何输入大小的输入,可以将输出尺寸指定为想要的维度,输入和输出特征的数目不会变化。

例如:

torch.nn.AdaptiveAvgPool1d(2)

torch.nn.AdaptiveAvgPool2d(2,2)

torch.nn.AdaptiveAvgPool3d(2,2,2)

你可能感兴趣的:(pytorch,神经网络)