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 – 量化輸入張量
kernel_size – 池化區域的大小。可以是單個數字或元組 (kD, kH, kW)
stride – 池化操作的步長。可以是單個數字或元組 (sD, sH, sW)。預設值:
kernel_sizepadding – 輸入兩側的隱式零填充。可以是單個數字或元組 (padD, padH, padW)。預設值: 0
ceil_mode – 如果為 True,計算輸出形狀時將使用 ceil 而非 floor。預設值:
Falsecount_include_pad – 如果為 True,將在平均計算中包含零填充。預設值:
Truedivisor_override – 如果指定,將用作除數,否則將使用池化區域的大小。預設值: None