我怎样才能开始为我的图层编写代码?

How can I start writing the code for my layer?

我看到研究人员正在为 Caffe 的原始版本添加一些功能,并根据需要使用这些层和功能,然后通过 Github 共享这些版本。如果我没记错的话,有两种方法: 1) 添加c++和Cuda版本的层后重新编译Caffe。 2) 为该功能编写 python 代码,并在 Caffe 中将其称为 python 层。

我想根据我的研究问题向Caffe 添加一个新层。我真的不知道我应该从哪个点开始写新层,我应该考虑哪些步骤。

我的问题是:

1)有没有什么文档或者学习资料可以用来写层?

2) 上述添加新层的方法中,哪种方式更受欢迎?

非常感谢任何帮助和指导

非常感谢

出于研究目的,对于"playing around",写一个python层通常更方便:省去了编译等工作
您可以在 "Python" 图层 .

上找到简短教程

另一方面,如果您想要更好的性能,您应该为您的层编写本机 C++ 代码。
你可以找到关于它的简短解释 here.