我有多个卷积层,具有相同的输出形状。
如何计算CAFFE中这些层输出的元素加权总和?
用于元素操作使用"Eltwise"
层。要获得"加权总和",您可以使用coeff
参数:
layer {
name: "wsum"
type: "Eltwise"
bottom: "x1"
bottom: "x2"
bottom: "x3"
top: "y"
eltwise_param {
operation: SUM
coeff: 0.5 # weight for x1
coeff: 0.3 # weight for x2
coeff: 0.2 # weight for x3
}
}
请注意,'coeff'
参数仅适用于SUM
操作。