我有一个nxmx16x1转换层,我想在整个通道中进行合并,因此结果的维度为nxmx1x1。
有什么建议吗?据我所知,汇集没有轴参数,对吗?
答案 0 :(得分:1)
你不需要axis
param。只需使用非均匀内核/步幅:
layer {
name: "pool16x1"
type: "Pool"
bottom: "input"
top: "output"
pooling_param {
kernel_size: 16
kernel_size: 1
stride: 16
stride: 1
# ...
}
}
应该做的伎俩...