【深度學習框架Caffe學習與應用】第五課 自定義神經層和數據輸入層

一、自定義神經層 1.創建新定義的頭文件，目錄在caffe/include/caffe/layers/my_neuron_layer.hpp   我複製了conv_layer.hpp文件，然後在該文件的內容上進行更改 如果只是需要CPU方法，可以註釋掉forward_gpu和backward_gpu 修改對照內容如下（左側是conv_layer.hpp，右側是my_neuron_layer.hpp

>>阅读原文<<