diff options
Diffstat (limited to 'layer')
| -rw-r--r-- | layer/affine.lua | 1 | ||||
| -rw-r--r-- | layer/sigmoid.lua | 14 |
2 files changed, 15 insertions, 0 deletions
diff --git a/layer/affine.lua b/layer/affine.lua index 221aacd..94e7497 100644 --- a/layer/affine.lua +++ b/layer/affine.lua @@ -11,6 +11,7 @@ function LinearTransParam:write(pfhandle) end function AffineLayer:__init(id, global_conf, ltp, bp) + self.id = id self.ltp = ltp self.bp = bp self.gconf = global_conf diff --git a/layer/sigmoid.lua b/layer/sigmoid.lua new file mode 100644 index 0000000..d69e9e3 --- /dev/null +++ b/layer/sigmoid.lua @@ -0,0 +1,14 @@ +local SigmoidLayer = nerv.class("nerv.SigmoidLayer", "nerv.Layer") + +function SigmoidLayer:__init(id, global_conf) + self.id = id + self.gconf = global_conf +end + +function SigmoidLayer:propagate(input, output) + output:sigmoid(input) +end + +function SigmoidLayer:back_propagate(next_bp_err, bp_err, input, output) + next_bp_err:sigmoid(bp_err, output) +end |