摘要:
对于layer层的cpp文件,你可以用LOG和printf、cout进行调试,cu文件不能使用LOG,可以使用cout,printf。 对于softmaxloss的layer层,既有cpp文件又有cu文件,其中cu文件只实现了前向和反向传播,cpp文件既实现了前向、反向,也实现了layer setu 阅读全文
摘要:
outer_num_:相当于batch_size dim: c*w*h spatial_dim(inner_num_):w*h softmax_loss.cpp的代码: 其实可以看出来count的只取前,不取后,(0, softmax_axis_)只取了0这一个轴 阅读全文