diff options
author | Qi Liu <liuq901@163.com> | 2016-06-21 15:42:22 +0800 |
---|---|---|
committer | Qi Liu <liuq901@163.com> | 2016-06-21 15:42:22 +0800 |
commit | 3856e63dab1b28aaec4133b6b0ec2a44ebf8cf46 (patch) | |
tree | ed20795f95854c72137bd7537a4097582e997fb3 /nerv/layer/softmax.lua | |
parent | fa6700c2caf2a857b95eb62a8dddbf2cfbad2145 (diff) |
complete bp of softmax layer
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 |