aboutsummaryrefslogtreecommitdiff
path: root/layer/affine.lua
diff options
context:
space:
mode:
authorcloudygoose <[email protected]>2015-06-02 11:14:18 +0800
committercloudygoose <[email protected]>2015-06-02 11:14:18 +0800
commit2bb2076fe12deb3bf6a38bd2e192dca06c7736e0 (patch)
tree8d0ae95e5474eb70d86a000ccc0b38df017af3e0 /layer/affine.lua
parent5e7fcdf4e5be450927764254d492d87349e4114e (diff)
parentd0a3e02d1a25a681ac78fd66aedf63f96636f6d2 (diff)
...
Merge remote-tracking branch 'upstream/master'
Diffstat (limited to 'layer/affine.lua')
-rw-r--r--layer/affine.lua9
1 files changed, 5 insertions, 4 deletions
diff --git a/layer/affine.lua b/layer/affine.lua
index d88813e..573b98d 100644
--- a/layer/affine.lua
+++ b/layer/affine.lua
@@ -1,13 +1,14 @@
-local LinearTransParam = nerv.class('nerv.LinearTransParam', 'nerv.Param')
-local BiasParam = nerv.class('nerv.BiasParam', 'nerv.LinearTransParam')
+local MatrixParam = nerv.class('nerv.MatrixParam', 'nerv.Param')
+local LinearTransParam = nerv.class('nerv.LinearTransParam', 'nerv.MatrixParam')
+local BiasParam = nerv.class('nerv.BiasParam', 'nerv.MatrixParam')
local AffineLayer = nerv.class('nerv.AffineLayer', 'nerv.Layer')
-function LinearTransParam:read(pcdata)
+function MatrixParam:read(pcdata)
self.trans = self.gconf.mat_type.new_from_host(
nerv.MMatrixFloat.load(pcdata))
end
-function LinearTransParam:write(pfhandle)
+function MatrixParam:write(pfhandle)
self.trans:new_to_host():save(pfhandle)
end