快捷方式

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]

kD timeskH×kWkD \ times kH \times kW 區域中應用 3D 平均池化操作,步長為 sD×sH×sWsD \times sH \times sW。輸出特徵的數量等於輸入平面的數量。

注意

輸入的量化引數會傳播到輸出。

引數
  • input – 量化輸入張量 (minibatch,in_channels,iH,iW)(\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

文件

查閱 PyTorch 的全面開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深度教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源