当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

Python torch.nn.AdaptiveAvgPool1d用法及代码示例


用法:

class torch.nn.AdaptiveAvgPool1d(output_size)

参数：

output_size-目标输出大小 。

在由多个输入平面组成的输入信号上应用一维自适应平均池化。

对于任何输入大小，输出大小为 。输出特征的数量等于输入平面的数量。

形状：

  • 输入： 或 。

  • 输出： 或 ，其中 。

例子

>>> # target output size of 5
>>> m = nn.AdaptiveAvgPool1d(5)
>>> input = torch.randn(1, 64, 8)
>>> output = m(input)

相关用法


注：本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.nn.AdaptiveAvgPool1d。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。