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) → 张量

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

详情和输出形状请参见 AvgPool3d。

Parameters

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

• kernel_size – 池化区域的大小。可以是单个数字或一个元组 (kT, kH, kW)

• 步幅 – 池化操作的步幅。可以是一个数字或一个元组 (sT, sH, sW)。默认值：kernel_size

• 填充 – 在输入的两边隐式地填充零。可以是一个数字或一个元组 (padT, padH, padW)，默认值：0

• ceil_mode – 当为True时，将在公式中使用ceil而不是floor来计算输出形状

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

• divisor_override – 如果指定，它将用作除数，否则将使用池化区域的大小。默认值：None