caffe CNN：跨渠道汇集

Question

我有一个 nxmx16x1 conv 层，我想跨通道进行池化，因此结果的维度为 nxmx1x1。

有什么建议吗？据我所知，池化没有轴参数，对吧？

Answer 1

您不需要 axis 参数。只需使用非均匀 kernel/stride:

layer {
  name: "pool16x1"
  type: "Pool"
  bottom: "input"
  top: "output"
  pooling_param {
    kernel_size: 16
    kernel_size: 1
    stride: 16
    stride: 1
    # ...
   }
}

应该可以解决问题...