torch.nn.functional.adaptive_avg_pool2d

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

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

有关详细信息和输出形状，请参阅 AdaptiveAvgPool2d。

参数:

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

返回类型:

张量