torch.nn.functional.adaptive_avg_pool2d

torch.nn.functional.adaptive_avg_pool2d(input, output_size)[源代码]

对由多个输入平面组成的输入信号进行二维自适应平均池化。

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

参数

output_size (None) – 指定目标输出的大小，可以是一个整数或者一个包含两个整数的元组。

返回类型

Tensor