From a309ce5e33b22030bcac348c63576187676abee3 Mon Sep 17 00:00:00 2001 From: Determinant Date: Mon, 1 Jun 2015 17:37:13 +0800 Subject: add expand_frm, rearrange_frm, scale_row --- layer/affine.lua | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) (limited to 'layer') 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 -- cgit v1.2.3