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$ 区域上以 $sT \times sH \times sW$ 步长应用 3D 平均池化操作。输出特征数量等于 $\lfloor\frac{\text{input planes}}{sT}\rfloor$。

注意

padding 最多应该是有效核大小的一半。

有关详细信息和输出形状，请参阅 AvgPool3d。

参数

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

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

• stride – 池化操作的步长。可以是单个数字或元组 (sT, sH, sW)。默认为 kernel_size

• padding – 输入两侧的隐式零填充。可以是单个数字或元组 (padT, padH, padW)，默认为 0

• ceil_mode – 如果为 True，则在计算输出形状的公式中使用 ceil 而不是 floor

• count_include_pad – 如果为 True，则在平均计算中包括零填充

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