代码之家  ›  专栏  ›  技术社区  ›  Nima

美国有线电视新闻网:汇集渠道

  •  0
  • Nima  · 技术社区  · 6 年前

    我有一个NXMX16X1 CONV层,我想做跨通道的汇集,所以结果具有NXMX1X1的维度。

    有什么建议吗?据我所知,池没有轴PARAM,对吗?

    1 回复  |  直到 6 年前
        1
  •  1
  •   Shai    6 年前

    你不需要 axis 帕拉姆只需使用非统一内核/步幅:

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

    应该做的把戏…