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 模型?