評價此頁

AdaptiveMaxPool1d#

class torch.nn.AdaptiveMaxPool1d(output_size, return_indices=False)[原始碼]#

對由多個輸入平面組成的輸入訊號應用 1D 自適應最大池化。

對於任何輸入大小,輸出大小均為 LoutL_{out}。輸出特徵的數量等於輸入通道的數量。

引數
  • output_size (Union[int, tuple[int]]) – 目標輸出大小 LoutL_{out}

  • return_indices (bool) – 如果為 True,則會返回包含輸出的索引。可用於傳遞給 nn.MaxUnpool1d。預設為 False

形狀
  • 輸入:(N,C,Lin)(N, C, L_{in})(C,Lin)(C, L_{in})

  • 輸出:(N,C,Lout)(N, C, L_{out})(C,Lout)(C, L_{out}),其中 Lout=output_sizeL_{out}=\text{output\_size}

示例

>>> # target output size of 5
>>> m = nn.AdaptiveMaxPool1d(5)
>>> input = torch.randn(1, 64, 8)
>>> output = m(input)
forward(input)[原始碼]#

執行前向傳播。