全局平均池化的计算过程

全局平均池化的计算过程如下: 假设输入的特征图大小为$H\times W\times C$,其中$H$表示高度,$W$表示宽度,$C$表示通道数。全局平均池化将输入特征图压缩成一个包含$C$个元素的向量,具体过程如下:

  1. 对于输入特征图的每个通道,计算该通道上所有元素的平均值,得到一个标量。这个标量就是该通道的输出值。
  2. 将每个通道的输出值按顺序排列,得到一个长度为$C$的向量。这个向量就是全局平均池化的输出。 以一个包含$3$个通道的$4\times 4$的特征图为例,全局平均池化的计算过程如下:
输入特征图:
[[[ 1,  2,  3,  4],
  [ 5,  6,  7,  8],
  [ 9, 10, 11, 12],
  [13, 14, 15, 16]],
 [[17, 18, 19, 20],
  [21, 22, 23, 24],
  [25, 26, 27, 28],
  [29, 30, 31, 32]],
 [[33, 34, 35, 36],
  [37, 38, 39, 40],
  [41, 42, 43, 44],
  [45, 46, 47, 48]]]
输出向量:
[22.5, 26.5, 30.5]

可以看到,对于每个通道,全局平均池化都将该通道上所有元素的平均值作为输出值。最终,将每个通道的输出值按顺序排列,得到一个包含$3$个元素的向量。