Fixed Tensor alloc for some modules (For CUDA)

Author: clementfarabet

CAddTable.lua

@@ -16,7 +16,7 @@ end
 
 function CAddTable:updateGradInput(input, gradOutput)
    for i=1,#input do
-      self.gradInput[i] = self.gradInput[i] or torch.Tensor()
+      self.gradInput[i] = self.gradInput[i] or input[1].new()
       self.gradInput[i]:resizeAs(input[i])
       self.gradInput[i]:copy(gradOutput)
    end

CDivTable.lua

@@ -13,8 +13,8 @@ function CDivTable:updateOutput(input)
 end
 
 function CDivTable:updateGradInput(input, gradOutput)
-   self.gradInput[1] = self.gradInput[1] or torch.Tensor()
-   self.gradInput[2] = self.gradInput[2] or torch.Tensor()
+   self.gradInput[1] = self.gradInput[1] or input[1].new()
+   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])
    return self.gradInput

CMulTable.lua

@@ -15,12 +15,13 @@ function CMulTable:updateOutput(input)
 end
 
 function CMulTable:updateGradInput(input, gradOutput)
-   local tout = torch.Tensor():resizeAs(self.output)
+   self.tout = self.tout or input[1].new()
+   self.tout:resizeAs(self.output)
    for i=1,#input do
-      self.gradInput[i] = self.gradInput[i] or torch.Tensor()
+      self.gradInput[i] = self.gradInput[i] or input[1].new()
       self.gradInput[i]:resizeAs(input[i]):copy(gradOutput)
-      tout:copy(self.output):cdiv(input[i])
-      self.gradInput[i]:cmul(tout)
+      self.tout:copy(self.output):cdiv(input[i])
+      self.gradInput[i]:cmul(self.tout)
    end
    return self.gradInput
 end

CSubTable.lua

@@ -13,8 +13,8 @@ function CSubTable:updateOutput(input)
 end
 
 function CSubTable:updateGradInput(input, gradOutput)
-   self.gradInput[1] = self.gradInput[1] or torch.Tensor()
-   self.gradInput[2] = self.gradInput[2] or torch.Tensor()
+   self.gradInput[1] = self.gradInput[1] or input[1].new()
+   self.gradInput[2] = self.gradInput[2] or input[1].new()
    self.gradInput[1]:resizeAs(input[1]):copy(gradOutput)
    self.gradInput[2]:resizeAs(input[1]):copy(gradOutput):mul(-1)
    return self.gradInput