From 3856e63dab1b28aaec4133b6b0ec2a44ebf8cf46 Mon Sep 17 00:00:00 2001 From: Qi Liu Date: Tue, 21 Jun 2016 15:42:22 +0800 Subject: complete bp of softmax layer --- nerv/layer/softmax.lua | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'nerv/layer/softmax.lua') diff --git a/nerv/layer/softmax.lua b/nerv/layer/softmax.lua index 81dcacc..6789ccc 100644 --- a/nerv/layer/softmax.lua +++ b/nerv/layer/softmax.lua @@ -32,7 +32,7 @@ function SoftmaxLayer:back_propagate(bp_err, next_bp_err, input, output) nbe:mul_elem(bp_err[1], output[1]) local offset = nbe:rowsum() nbe:copy_from(bp_err[1]) - nbe:add_row(offset, -1.0) + nbe:add_col(offset, -1.0) nbe:mul_elem(nbe, output[1]) end -- cgit v1.2.3