我有一个维度nXmx16x1的转换层,我想计算16个滤波器的平均值或总和,因此结果是1个暗淡的nxmx1x1滤波器。
类似于matlab中的均值(A,3)
答案 0 :(得分:1)
使用常量1x1 "Conv"
过滤器:
layer {
name: "avg"
type: "Convolution"
bottom: "input"
top: "output"
param { lr_mult: 0 decay_mult: 0 } # fix this layer, do not train
convolution_param {
num_output: 1
kernel_size: 1
bias: false # no bias
weight_filler: { type: "constant" value: 0.0625 } # 1/16.
}
}