SoftmaxLayer and SoftmaxwithLossLayer 代码解读
Wang Xiao
先来看看 SoftmaxWithLoss 在prototext文件中的定义:
再看SoftmaxWithLossLayer的.cpp文件:
接下来是对输入数据进行 reshape 操作:
SoftmaxLayer and SoftmaxwithLossLayer 代码解读
Wang Xiao
先来看看 SoftmaxWithLoss 在prototext文件中的定义:
再看SoftmaxWithLossLayer的.cpp文件:
接下来是对输入数据进行 reshape 操作: