1 files changed, 37 insertions, 0 deletions
diff --git a/nerv/layer/rnn.lua b/nerv/layer/rnn.lua
new file mode 100644
index 0000000..a93530f
--- /dev/null
+++ b/nerv/layer/rnn.lua
@@ -0,0 +1,37 @@
+local RNNLayer = nerv.class('nerv.RNNLayer', 'nerv.GraphLayer')
+
+function RNNLayer:__init(id, global_conf, layer_conf)
+    self.id = id
+    self.dim_in = layer_conf.dim_in
+    self.dim_out = layer_conf.dim_out
+    self.gconf = layer_conf.gconf
+    self:check_dim_len(1, 1)
+
+    local din = layer_conf.dim_in[1]
+    local dout = layer_conf.dim_out[1]
+
+    local pr = layer_conf.pr
+    if pr == nil then
+        pr = nerv.ParamRepo()
+    end
+
+    local layers = {
+        ['nerv.AffineLayer'] = {
+            main = {dim_in = {din, dout}, dim_out = {dout}, pr = pr},
+        },
+        ['nerv.SigmoidLayer'] = {
+            sigmoid = {dim_in = {dout}, dim_out = {dout}},
+        },
+    }
+
+    local layer_repo = nerv.LayerRepo(layers, pr, global_conf)
+
+    local connections = {
+        {'<input>[1]', 'main[1]', 0},
+        {'main[1]', 'sigmoid[1]', 0},
+        {'sigmoid[1]', 'main[2]', 0},
+        {'sigmoid[1]', '<output>[1]', 1},
+    }
+
+    self:graph_init(layer_repo, connections)
+end