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)

forward(input)

執行前向傳播。

返回型別

張量