aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--nerv/layer/affine.lua20
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')