AvgPool2d¶

class AvgPool2d(kernel_size, stride=None, padding=0, mode='average_count_exclude_padding', **kwargs)[源代码]¶

对输入数据进行2D平均池化。

例如，给定形状为 \((N, C, H, W)\) 的输入以及形为 \((kH, kW)\) 的 kernel_size ，该层产生形状为 \((N, C, H_{out}, W_{out})\) 的输出。生成过程描述如下：

\[out(N_i, C_j, h, w) = \frac{1}{kH * kW} \sum_{m=0}^{kH-1} \sum_{n=0}^{kW-1} input(N_i, C_j, stride[0] \times h + m, stride[1] \times w + n)\]

若 padding 非零，则输入数据会被隐式地在两边用零值进行填充（pad) padding 个点。

参数

kernel_size (Union[int, Tuple[int, int]]) – 窗的大小。
stride (Union[int, Tuple[int, int], None]) – 窗的步长。默认值是 kernel_size。
padding (Union[int, Tuple[int, int]]) – 对两边进行隐式的零值填充尺寸。
mode (str) – whether to count padding values. “average” mode will do counting and “average_count_exclude_padding” mode won’t do counting. Default: “average_count_exclude_padding”

SlidingWindowTranspose

MaxPool2d