一维自适应最大池化

classtorch.nn.AdaptiveMaxPool1d(output_size, return_indices=False)[源代码]

对由多个输入平面组成的一维输入信号进行自适应最大池化操作。

输出大小为$L_{out}$,适用于任意输入大小。输出特征的数量与输入平面的数量相等。

参数
  • output_size (Union[int, Tuple[int]]) – 目标输出大小 $L_{out}$.

  • return_indices (bool) – 如果为True,将返回索引和输出。传递给 nn.MaxUnpool1d 时有用。默认值:False

形状:
  • 输入格式为: $(N, C, L_{in})$$(C, L_{in})$

  • 输出: $(N, C, L_{out})$$(C, L_{out})$,其中 $L_{out}$ 等于 \text{output\_size}

示例

>>> # target output size of 5
>>> m = nn.AdaptiveMaxPool1d(5)
>>> input = torch.randn(1, 64, 8)
>>> output = m(input)
本页目录