我有多个具有相同输出形状的卷积层。
如何在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
操作。