PyTorch
torch / nn / torch.nn
AdaptiveAvgPool1d¶
-
class
torch.nn.
AdaptiveAvgPool1d
(output_size: Union[T, Tuple[T, ...]])[source]¶ Applies a 1D adaptive average pooling over an input signal composed of several input planes.
The output size is H, for any input size. The number of output features is equal to the number of input planes.
- Parameters
output_size – the target output size H
Examples
>>> # target output size of 5 >>> m = nn.AdaptiveAvgPool1d(5) >>> input = torch.randn(1, 64, 8) >>> output = m(input)
此页内容是否对您有帮助
感谢反馈!