AdaptiveAvgPool1d

class torch.nn.modules.pooling.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)

forward(input)

執行前向傳播。

返回型別

張量