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)[source][source]¶
在 区域通过步长 步应用 3D 平均池化操作。输出特征的数量等于输入平面的数量。
注意
输入量化参数会传播到输出。
- 参数:
input – 量化输入张量
核大小 – 池化区域的尺寸。可以是单个数字或元组(kD, kH, kW)
步长 – 池化操作的步长。可以是单个数字或元组(sD, sH, sW)。默认:
kernel_size
填充 – 输入两侧的隐式零填充。可以是单个数字或元组(padD, padH, padW)。默认:0
ceil_mode – 当为 True 时,将使用 ceil 代替 floor 来计算输出形状的公式。默认:
False
count_include_pad – 当为 True 时,将包括零填充在平均计算中。默认:
True
divisor_override – 如果指定,则用作除数,否则使用池化区域的大小。默认:None