快捷键

torch.nn.functional.max_pool1d

torch.nn.functional.max_pool1d(input, kernel_size, stride=None, padding=0, dilation=1, ceil_mode=False, return_indices=False)[source]

在由多个输入平面组成的输入信号上应用 1D 最大池化。

注意

ceil_modereturn_indices 的顺序与 MaxPool1d 中看到的不同,将在未来的版本中更改。

详细内容请见 MaxPool1d

参数:
  • 输入 - 形状为 (minibatch,in_channels,iW)(\text{minibatch} , \text{in\_channels} , iW) 的输入张量,minibatch 维度可选。

  • 核心大小 – 窗口大小。可以是单个数字或元组(kW,)

  • 步长 – 窗口的步长。可以是单个数字或元组(sW,)。默认: kernel_size

  • 填充 - 在两侧添加隐式负无穷填充,必须大于等于 0 且小于等于 kernel_size / 2。

  • 扩展率 - 滑动窗口内元素之间的步长,必须大于 0。

  • ceil_mode – 如果 True ,将使用向上取整而不是向下取整来计算输出形状。这确保输入张量中的每个元素都被滑动窗口覆盖。

  • return_indices – 如果 True ,将返回 argmax 及其最大值。对于 torch.nn.functional.max_unpool1d 很有用。


© 版权所有 PyTorch 贡献者。

使用 Sphinx 构建,并使用 Read the Docs 提供的主题。

文档

PyTorch 的全面开发者文档

查看文档

教程

深入了解初学者和高级开发者的教程

查看教程

资源

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

查看资源