Caffe 学习:Eltwise层

Posted 坤元居士

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了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)都要给出!

以上是关于Caffe 学习:Eltwise层的主要内容,如果未能解决你的问题,请参考以下文章

如何将 caffe prototxt 转换为 pytorch 模型?

Caffe 学习:Crop 层

转 Caffe学习系列:视觉层(Vision Layers)及参数

转 Caffe学习系列:数据层及参数

转 Caffe学习系列:其它常用层及参数

Caffe学习5-用C++自定义层以及可视化结果