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$ 區域上應用 3D 平均池化 (average-pooling) 運算，步長為 $sT \times sH \times sW$ 。輸出特徵的數量等於 $\lfloor\frac{\text{input planes}}{sT}\rfloor$。

有關詳細資訊和輸出形狀，請參閱 AvgPool3d。

參數

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

• kernel_size – 池化區域的大小。可以是單個數字或一個元組 (kT, kH, kW)

• stride – 池化運算的步長。可以是單個數字或一個元組 (sT, sH, sW)。預設值：kernel_size

• padding – 輸入兩側的隱式零填充 (zero padding)。可以是單個數字或一個元組 (padT, padH, padW)，預設值：0

• ceil_mode – 如果為 True，將在公式中使用 ceil 而不是 floor 來計算輸出形狀

• count_include_pad – 如果為 True，將在平均計算中包含零填充

• divisor_override – 如果指定，它將用作除數，否則將使用池化區域的大小。預設值：None