Caffe 学习:Eltwise层

 

  Eltwise层的操作有三个:product(点乘), sum(相加减) 和 max(取大值),其中sum是默认操作。

 

  假设输入(bottom)为A和B,如果要实现element_wise的A+B,即A和B的对应元素相加,prototxt文件如下:

复制代码
layer 
{
  name: "eltwise_layer"
  type: "Eltwise"
  bottom: "A"
  bottom: "B"
  top: "diff"
  eltwise_param {
    operation: SUM
  }
}​
复制代码

 

  如果实现A-B,则prototxt为:

复制代码
layer 
{
  name: "eltwise_layer"
  type: "Eltwise"
  bottom: "A"
  bottom: "B"
  top: "diff"
  eltwise_param {
    operation: SUM
    coeff: 1
    coeff: -1
  }
}​
复制代码

  其中A和B的系数(coefficient)都要给出!

posted @   坤元居士  阅读(45585)  评论(1编辑  收藏  举报
努力加载评论中...
点击右上角即可分享
微信分享提示