diff options
| author | Determinant <ted.sybil@gmail.com> | 2015-05-26 14:06:52 +0800 |
|---|---|---|
| committer | Determinant <ted.sybil@gmail.com> | 2015-05-26 14:06:52 +0800 |
| commit | 81bf2d653902860c5d28ccade19ac6e1fd56acaf (patch) | |
| tree | 831a25c84332ac0839dbe498f61620ea634170e0 /layer/affine.lua | |
| parent | 8c5246a8794011ca0c25f15643771f031d258594 (diff) | |
add layer and param
Diffstat (limited to 'layer/affine.lua')
| -rw-r--r-- | layer/affine.lua | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/layer/affine.lua b/layer/affine.lua new file mode 100644 index 0000000..d5c50fc --- /dev/null +++ b/layer/affine.lua @@ -0,0 +1,11 @@ +local LinearTransParam = nerv.class('nerv.LinearTransParam', 'nerv.Param') +local BiasParam = nerv.class('nerv.BiasParam', 'nerv.LinearTransParam') +local AffineLayer = nerv.class('nerv.AffineLayer', 'nerv.Layer') + +function LinearTransParam:read(pcdata) + self.trans = nerv.CuMatrixFloat.new_from_host(nerv.MMatrixFloat.load(pcdata)) +end + +function LinearTransParam:write(pfhandle) + self.trans:new_to_host():save(pfhandle) +end |