diff options
Diffstat (limited to 'layer')
| -rw-r--r-- | layer/affine.lua | 14 | ||||
| -rw-r--r-- | layer/init.lua | 2 |
2 files changed, 15 insertions, 1 deletions
diff --git a/layer/affine.lua b/layer/affine.lua index d5c50fc..5f1b4ce 100644 --- a/layer/affine.lua +++ b/layer/affine.lua @@ -9,3 +9,17 @@ end function LinearTransParam:write(pfhandle) self.trans:new_to_host():save(pfhandle) end + +function AffineLayer:__init(id, ltp, bp) + self.ltp = ltp + self.bp = bp +end + +function nerv.AffineLayer:update(input, output) +end + +function nerv.AffineLayer:propagate(input, output) +end + +function nerv.AffineLayer:back_propagate(input, output) +end diff --git a/layer/init.lua b/layer/init.lua index c57a405..62a2924 100644 --- a/layer/init.lua +++ b/layer/init.lua @@ -24,7 +24,7 @@ end local Layer = nerv.class('nerv.Layer') -function nerv.Layer:_init(param) +function nerv.Layer:_init(id, ...) nerv.error_method_not_implemented() end |