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)都要給出!


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM