SoftmaxLayer and SoftmaxwithLossLayer 代碼解讀
Wang Xiao
先來看看 SoftmaxWithLoss 在prototext檔案中的定義:
再看SoftmaxWithLossLayer的.cpp檔案:
接下來是對輸入資料進行 reshape 操作:
SoftmaxLayer and SoftmaxwithLossLayer 代碼解讀
Wang Xiao
先來看看 SoftmaxWithLoss 在prototext檔案中的定義:
再看SoftmaxWithLossLayer的.cpp檔案:
接下來是對輸入資料進行 reshape 操作: