开发者社区

文档建议反馈控制台

最新优惠活动

文章/答案/技术大牛

发布

社区首页 >专栏 >caffe源码分析-layer_factory

caffe源码分析-layer_factory

bear_fish

发布于 2019-02-25 11:51:52

4950

发布于 2019-02-25 11:51:52

举报

文章被收录于专栏：用户2442861的专栏用户2442861的专栏

caffe中有许多的layer,在net中创建连接layer是通过工厂模式的方式创建，而不是每一个new然后连接。在net.cpp中创建layer方式如下：

layers_.push_back(LayerRegistry<Dtype>::CreateLayer(layer_param));

下面简要分析下layer_factory

结合宏定义以及模板特化创建的对应的注册类：

使用如下,例如输入层：

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2018年10月07日，如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自作者个人站点/博客前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体同步曝光计划，欢迎热爱写作的你一起参与！

评论

登录后参与评论

0 条评论

热度

最新

LV.