torch.nn.functional.avg_pool3d

torch.nn.functional.avg_pool3d(input, kernel_size, stride=None, padding=0, ceil_mode=False, count_include_pad=True, divisor_override=None) → Tensor

在 $kT \times kH \times kW$ 区域内应用 3D 平均池化操作，步长为 $sT \times sH \times sW$ 步。输出特征的数量等于 $\lfloor\frac{\text{input planes}}{sT}\rfloor$ 。

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

参数:

• 输入 – 输入张量 $(\text{minibatch} , \text{in\_channels} , iT \times iH , iW)$

• 核大小 – 池化区域的尺寸。可以是单个数字或一个元组（kT, kH, kW）

• 步长 – 池化操作的步长。可以是单个数字或一个元组（sT, sH, sW）。默认： kernel_size

• 填充 – 输入两侧的隐式零填充。可以是单个数字或一个元组（padT, padH, padW），默认：0

• ceil_mode – 当为 True 时，将使用向上取整而不是向下取整的公式来计算输出形状

• count_include_pad – 当为 True 时，将包括零填充在平均计算中

• 如果指定，则使用该除数，否则使用池化区域的大小。默认：无