diff options
author | Determinant <ted.sybil@gmail.com> | 2016-03-17 10:54:06 +0800 |
---|---|---|
committer | Determinant <ted.sybil@gmail.com> | 2016-03-17 10:54:06 +0800 |
commit | 941a3a6ee81587568a7f8e7458e5c5f9a8595342 (patch) | |
tree | fb3bb0d210d6cd347f60098faeeeb8afc70fd6c5 /nerv/layer/graph.lua | |
parent | 00211587988024e34370e7ec2ee1cb1c8632f116 (diff) | |
parent | 284d9f3fe3f7017e677fac03b2bc134892875515 (diff) |
Merge branch 'liuq901-master'
Diffstat (limited to 'nerv/layer/graph.lua')
-rw-r--r-- | nerv/layer/graph.lua | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/nerv/layer/graph.lua b/nerv/layer/graph.lua index 68d5f51..ddbc85e 100644 --- a/nerv/layer/graph.lua +++ b/nerv/layer/graph.lua @@ -65,7 +65,17 @@ function GraphLayer:discover(id, layer_repo) return ref end +local function reverse(connections) + for i = 1, #connections do + connections[i][3] = connections[i][3] * -1 + end +end + function GraphLayer:graph_init(layer_repo, connections) + if self.lconf.reversed then + reverse(connections) + end + local layers = {} layers['<input>'] = { inputs = {}, |