avg_pool3d

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

在 $kD \ times kH \times kW$ 区域上应用 3D 平均池化操作，步长为 $sD \times sH \times sW$。 输出特征的数量等于输入平面的数量。

注意

输入量化参数会传播到输出。

参数

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

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

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

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

• ceil_mode – 如果为 True，则在公式中使用 ceil 而不是 floor 来计算输出形状。 默认值： False

• count_include_pad – 如果为 True，则在平均计算中包含零填充。 默认值： True

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