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.