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

08-19 20:11