add add_row for cumatrix

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