深度学习入门第五关:Global Average Pooling 到底是什么?为什么能把 [8,128,8,8] 变成 [8,128]?
从入门到入神|PyTorch 基础系列 05
上一篇我们讲了 Flatten:
[8,128,8,8]
↓
Flatten
↓
[8,8192]但在很多 CNN 中,还会看到这样的代码:
nn.AdaptiveAvgPool2d((1, 1))然后 Tensor 变成:
[8,128,8,8]
↓
[8,128,1,…
2026/8/16 1:43:33