diff options
Diffstat (limited to 'nerv/layer/softmax.lua')
-rw-r--r-- | nerv/layer/softmax.lua | 2 |
1 files changed, 1 insertions, 1 deletions
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 |