AdaptiveAvgPool1d

class torch.nn.AdaptiveAvgPool1d(output_size)

在由多個輸入平面組成的輸入訊號上應用一維自適應平均池化。

輸出大小為 $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)