快捷方式

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×kH×kWkT \times kH \times kW 区域内应用 3D 平均池化操作,步长为 sT×sH×sWsT \times sH \times sW。输出特征的数量等于 input planessT\lfloor\frac{\text{input planes}}{sT}\rfloor.

有关详细信息和输出形状,请参见 AvgPool3d

参数
  • input – 输入张量 (minibatch,in_channels,iT×iH,iW)(\text{minibatch} , \text{in\_channels} , iT \times iH , iW)

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

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

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

  • ceil_mode – 当为 True 时,将在计算输出形状的公式中使用 ceil 而不是 floor

  • count_include_pad – 当为 True 时,将在平均计算中包含零填充

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

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源