PyTorch nn.AdaptiveAvgPool1d() 计算方法详解

nn.AdaptiveAvgPool1d() 的计算方法如下：

1. 输入一个形状为 (batch_size, channels, sequence_length) 的张量。

2. 在 sequence_length 维度上进行平均池化，即对每个通道的所有元素取平均值，得到一个形状为 (batch_size, channels, 1) 的张量。

3. 如果输出的目标形状为 (target_sequence_length)，则将上一步得到的 1 维张量在 sequence_length 维度上进行插值，将其扩展到目标形状，并返回结果。

例如，如果输入张量为 (batch_size, channels, 10)，输出目标形状为 (5)，则 AdaptiveAvgPool1d 将对每个通道的 10 个元素取平均值，得到一个形状为 (batch_size, channels, 1) 的张量，然后在 sequence_length 维度上对该张量进行插值，将其扩展到形状为 (batch_size, channels, 5) 的张量，并返回结果。