caffe代碼閱讀5:Layer的實現細節

一、Layer的作用簡介 Layer實際上定義了Layer的基本操作,即初始化層、前向傳播和反向傳播。在前向傳播中根據bottom blob得到top blob,反向傳播則根據top反傳到bottom。而且在前傳的時候還可以計算loss,一般來說只有最後一層纔會計算loss,雖然每個層都有計算loss的功能。Layer類在沒有實現GPU前傳和反傳的時候會自動使用CPU的實現。下面給出Layer類的
相關文章
相關標籤/搜索