美文网首页
【caffe添加新层】通道复制层ChCopy

【caffe添加新层】通道复制层ChCopy

作者: Scharlie | 来源:发表于2017-11-20 21:08 被阅读93次

该层的作用是将n*1*h*w维度的数据复制成n*copynum*h*w维的数据。

1.首先是.hpp和.cpp文件的编写,源码如下

《.hpp文件》

.hpp

《.cpp文件》

层setup以及reshape部分 前向传播部分 反向传播及“注册”部分

《caffe.proto文件修改》:在message中声明chcopy参数,以及重用该参数

进入caffe根目录,

make clean

make all

make all -j8

2.使用时,写好相应的prototxt文件,即可

相关文章

网友评论

      本文标题:【caffe添加新层】通道复制层ChCopy

      本文链接:https://www.haomeiwen.com/subject/xlebvxtx.html