correct the updateGradInput logic of module.

We want to make sure that if a node has only a single child, then we pass whwtever the
output of updateGradInput is, else, we select an pass to tthe corresponding child.

2 files changed, 2 insertions, 2 deletions

diff --git a/README.md b/README.md
index 6ad0282..f572553 100644
--- a/README.md
+++ b/README.md
@@ -87,7 +87,7 @@ graph.dot(gmod.fg,'Big MLP')
 	m:add(nn.SplitTable(1))
 	m:add(nn.ParallelTable():add(nn.Linear(10,20)):add(nn.Linear(10,30)))
 	input = nn.Identity()()
-	input1,input2 = m(input,2)
+	input1,input2 = m(input):split(2)
 	m3 = nn.JoinTable(1)({input1,input2})
 
 	g = nn.gModule({input},{m3})
diff --git a/gmodule.lua b/gmodule.lua
index b1b1e79..4b4f5f1 100644
--- a/gmodule.lua
+++ b/gmodule.lua
@@ -226,7 +226,7 @@ function gModule:updateGradInput(input,gradOutput)
 				child.data.gradOutput = child.data.gradOutput or {}
 				local mapindex = node.data.mapindex[child.data]
 				local gi
-				if istable(gradInput) and istable(input) then
+				if #node.children ~= 1 then --istable(gradInput) and istable(input) then
 					gi = gradInput[mapindex]
 				else
 					gi = gradInput