torch.nn.functional.adaptive_avg_pool2d

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

对由多个输入通道组成的输入信号应用二维自适应平均池化。

请参阅 AdaptiveAvgPool2d 查看详细信息和输出形状。

参数

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

返回类型

张量