天天看點

SoftmaxLayer and SoftmaxwithLossLayer 代碼解讀

SoftmaxLayer and SoftmaxwithLossLayer 代碼解讀

Wang Xiao

  先來看看 SoftmaxWithLoss 在prototext檔案中的定義:

  再看SoftmaxWithLossLayer的.cpp檔案:

  

  接下來是對輸入資料進行 reshape 操作: 

繼續閱讀