AdaptiveAvgPool1d

class torch.nn.AdaptiveAvgPool1d(output_size)

將 1D 自適應平均池化應用於由多個輸入平面組成的輸入信號。

對於任何輸入大小，輸出大小皆為 $L_{out}$。輸出特徵的數量等於輸入平面的數量。

參數

output_size (Union[int, Tuple[int]]) – 目標輸出大小 $L_{out}$。

形狀

• 輸入: $(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.AdaptiveAvgPool1d(5)
>>> input = torch.randn(1, 64, 8)
>>> output = m(input)