torch.nn.functional.adaptive_avg_pool2d

torch.nn.functional.adaptive_avg_pool2d(input, output_size)

对由多个输入平面组成的输入信号应用 2D 自适应平均池化。

详细信息和输出形状请见 AdaptiveAvgPool2d 。

参数:

output_size (None) – 目标输出大小（单个整数或双整数元组）

返回类型:

张量