From 2ea3e139af91eb894d904d7a956e28619b1a70f6 Mon Sep 17 00:00:00 2001
From: Qi Liu <liuq901@163.com>
Date: Tue, 1 Mar 2016 20:00:53 +0800
Subject: network init complete

---
 nerv/layer/rnn.lua | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

(limited to 'nerv/layer/rnn.lua')

diff --git a/nerv/layer/rnn.lua b/nerv/layer/rnn.lua
index 8816891..806ac58 100644
--- a/nerv/layer/rnn.lua
+++ b/nerv/layer/rnn.lua
@@ -22,6 +22,9 @@ function RNNLayer:__init(id, global_conf, layer_conf)
         ['nerv.SigmoidLayer'] = {
             sigmoid = {dim_in = {dout}, dim_out = {dout}},
         },
+        ['nerv.DuplicateLayer'] = {
+            dup = {dim_in = {dout}, dim_out = {dout, dout}},
+        }
     }
 
     local layer_repo = nerv.LayerRepo(layers, pr, global_conf)
@@ -29,8 +32,9 @@ function RNNLayer:__init(id, global_conf, layer_conf)
     local connections = {
         {'<input>[1]', 'main[1]', 0},
         {'main[1]', 'sigmoid[1]', 0},
-        {'sigmoid[1]', 'main[2]', 1},
-        {'sigmoid[1]', '<output>[1]', 0},
+        {'sigmoid[1]', 'dup[1]', 0},
+        {'dup[1]', 'main[2]', 1},
+        {'dup[2]', '<output>[1]', 0},
     }
 
     self:graph_init(layer_repo, connections)
-- 
cgit v1.2.3