fixing table modules to return correct number of gradInputs

Author: soumith

CAddTable.lua

@@ -20,5 +20,10 @@ function CAddTable:updateGradInput(input, gradOutput)
       self.gradInput[i]:resizeAs(input[i])
       self.gradInput[i]:copy(gradOutput)
    end
+
+   for i=#input+1, #self.gradInput do
+       self.gradInput[i] = nil
+   end
+
    return self.gradInput
 end

CDivTable.lua

@@ -17,5 +17,10 @@ function CDivTable:updateGradInput(input, gradOutput)
    self.gradInput[2] = self.gradInput[2] or input[1].new()
    self.gradInput[1]:resizeAs(input[1]):copy(gradOutput):cdiv(input[2])
    self.gradInput[2]:resizeAs(input[2]):zero():addcdiv(-1,self.gradInput[1],input[2]):cmul(input[1])
+
+   for i=#input+1, #self.gradInput do
+       self.gradInput[i] = nil
+   end
+
    return self.gradInput
 end

CMulTable.lua

@@ -23,6 +23,11 @@ function CMulTable:updateGradInput_efficient(input, gradOutput)
       self.tout:copy(self.output):cdiv(input[i])
       self.gradInput[i]:cmul(self.tout)
    end
+
+   for i=#input+1, #self.gradInput do
+       self.gradInput[i] = nil
+   end
+
    return self.gradInput
 end
 
@@ -36,6 +41,10 @@ function CMulTable:updateGradInput(input, gradOutput)
          end
       end
    end
+
+   for i=#input+1, #self.gradInput do
+       self.gradInput[i] = nil
+   end
+
    return self.gradInput
 end
-

CSubTable.lua

@@ -17,5 +17,10 @@ function CSubTable:updateGradInput(input, gradOutput)
    self.gradInput[2] = self.gradInput[2] or input[1].new()
    self.gradInput[1]:resizeAs(input[1]):copy(gradOutput)
    self.gradInput[2]:resizeAs(input[2]):copy(gradOutput):mul(-1)
+
+   for i=#input+1, #self.gradInput do
+       self.gradInput[i] = nil
+   end
+
    return self.gradInput
 end