摘要: 对于layer层的cpp文件,你可以用LOG和printf、cout进行调试,cu文件不能使用LOG,可以使用cout,printf。 对于softmaxloss的layer层,既有cpp文件又有cu文件,其中cu文件只实现了前向和反向传播,cpp文件既实现了前向、反向,也实现了layer setu 阅读全文
posted @ 2018-02-27 15:51 有梦就要去实现他 阅读(597) 评论(0) 推荐(0) 编辑
摘要: outer_num_:相当于batch_size dim: c*w*h spatial_dim(inner_num_):w*h softmax_loss.cpp的代码: 其实可以看出来count的只取前,不取后,(0, softmax_axis_)只取了0这一个轴 阅读全文
posted @ 2018-02-27 15:25 有梦就要去实现他 阅读(604) 评论(0) 推荐(0) 编辑