美文网首页
【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