1 files changed, 31 insertions, 0 deletions
diff --git a/nerv/main.lua b/nerv/main.lua
new file mode 100644
index 0000000..85e291c
--- /dev/null
+++ b/nerv/main.lua
@@ -0,0 +1,31 @@
+print 'Hello'
+
+local global_conf = {
+    cumat_type = nerv.CuMatrixFloat,
+    param_random = function() return 0 end,
+}
+
+local layer_repo = nerv.LayerRepo(
+    {
+        ['nerv.RNNLayer'] = {
+            rnn = {dim_in = {23}, dim_out = {26}},
+        },
+        ['nerv.AffineLayer'] = {
+            input = {dim_in = {20}, dim_out = {23}},
+            output = {dim_in = {26, 79}, dim_out = {79}},
+        },
+        ['nerv.SigmoidLayer'] = {
+            sigmoid = {dim_in = {23}, dim_out = {23}},
+        },
+    }, nerv.ParamRepo(), global_conf)
+
+local connections = {
+    {'<input>[1]', 'input[1]', 0},
+    {'input[1]', 'sigmoid[1]', 0},
+    {'sigmoid[1]', 'rnn[1]', 0},
+    {'rnn[1]', 'output[1]', 0},
+    {'output[1]', 'output[2]', 1},
+    {'output[1]', '<output>[1]', 0},
+}
+
+local network = nerv.GraphLayer('network', global_conf, {dim_in = {20}, dim_out = {79}, layer_repo = layer_repo, connections = connections})