torch.nn.functional.avg_pool2d¶
- torch.nn.functional.avg_pool2d(input, kernel_size, stride=None, padding=0, ceil_mode=False, count_include_pad=True, divisor_override=None) Tensor ¶
在 区域上应用 2D 平均池化操作,步长为 步长。输出特征的数量等于输入平面的数量。
有关详细信息和输出形状,请参见
AvgPool2d
。- 参数
input – 输入张量
kernel_size – 池化区域的大小。可以是单个数字或元组 (kH, kW)
stride – 池化操作的步长。可以是单个数字或元组 (sH, sW)。默认值:
kernel_size
padding – 输入两侧的隐式零填充。可以是单个数字或元组 (padH, padW)。默认值:0
ceil_mode – 当为 True 时,将在计算输出形状的公式中使用 ceil 而不是 floor。默认值:
False
count_include_pad – 当为 True 时,将在平均计算中包含零填充。默认值:
True
divisor_override – 如果指定,它将用作除数,否则将使用池化区域的大小。默认值:None