AdaptiveMaxPool1d

class torch.nn.modules.pooling.AdaptiveMaxPool1d(output_size, return_indices=False)

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

對於任何輸入大小，輸出大小均為 $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)

forward(input)

執行前向傳播。