aboutsummaryrefslogtreecommitdiff
path: root/nerv/layer/init.lua
diff options
context:
space:
mode:
authorQi Liu <liuq901@163.com>2016-02-29 17:46:09 +0800
committerQi Liu <liuq901@163.com>2016-02-29 17:46:09 +0800
commit77b558898a2a29097d8697a59a7d23cd2a52975f (patch)
tree06bab2379224a6d06bd6b9c60468597e1fbe6e1e /nerv/layer/init.lua
parent550680eacd00555817df19d2b59a20a92df77c42 (diff)
graph layer complete
Diffstat (limited to 'nerv/layer/init.lua')
-rw-r--r--nerv/layer/init.lua10
1 files changed, 10 insertions, 0 deletions
diff --git a/nerv/layer/init.lua b/nerv/layer/init.lua
index 54f33ae..5e3395c 100644
--- a/nerv/layer/init.lua
+++ b/nerv/layer/init.lua
@@ -70,6 +70,14 @@ function Layer:get_dim()
return self.dim_in, self.dim_out
end
+function Layer:set_attr(name, value)
+ self[name] = value
+end
+
+function Layer:get_sublayer(id)
+ nerv.error('primitive layer does not have sublayers.')
+end
+
function Layer:find_param(pid_list, lconf, gconf, p_type, p_dim)
if type(pid_list) == "string" then
pid_list = {pid_list}
@@ -101,6 +109,7 @@ function Layer:find_param(pid_list, lconf, gconf, p_type, p_dim)
return p
end
+nerv.include('graph.lua')
nerv.include('affine.lua')
nerv.include('sigmoid.lua')
nerv.include('tanh.lua')
@@ -115,6 +124,7 @@ nerv.include('lstm.lua')
nerv.include('lstm_gate.lua')
nerv.include('dropout.lua')
nerv.include('gru.lua')
+nerv.include('rnn.lua')
-- The following lines are for backward compatibility, and will be removed in
-- the future. The use of these names are deprecated.