diff options
| -rw-r--r-- | nerv/layer/affine.lua | 20 |
1 files changed, 10 insertions, 10 deletions
diff --git a/nerv/layer/affine.lua b/nerv/layer/affine.lua index 2dd2dc0..b4358ca 100644 --- a/nerv/layer/affine.lua +++ b/nerv/layer/affine.lua @@ -1,16 +1,6 @@ --- Contains parameter and layer classes related to linear (or affine) -- transform. ---- The class for linear transform parameter. --- @type nerv.LinearTransParam - -local LinearTransParam = nerv.class('nerv.LinearTransParam', 'nerv.MatrixParam') - ---- The class for bias parameter (currently implemented as a one-row matrix). --- @type nerv.BiasParam - -local BiasParam = nerv.class('nerv.BiasParam', 'nerv.MatrixParam') - --- The class for all matrix-based parameters. The class has a single matrix -- which can be accessed by `self.trans`. -- @type nerv.MatrixParam @@ -174,3 +164,13 @@ function AffineLayer:get_params() end return pr end + +--- The class for linear transform parameter. +-- @type nerv.LinearTransParam + +local LinearTransParam = nerv.class('nerv.LinearTransParam', 'nerv.MatrixParam') + +--- The class for bias parameter (currently implemented as a one-row matrix). +-- @type nerv.BiasParam + +local BiasParam = nerv.class('nerv.BiasParam', 'nerv.MatrixParam') |