diff options
author | Clement Farabet <clement.farabet@gmail.com> | 2012-10-22 23:33:13 +0400 |
---|---|---|
committer | Clement Farabet <clement.farabet@gmail.com> | 2012-10-22 23:33:13 +0400 |
commit | 1d5b59b9f202f64c854d2267b79e97596a3c1d26 (patch) | |
tree | 2d064db8666b373856c9059a3425ecd743d8bf31 /CDivTable.lua | |
parent | b8addac4e7fe7171e42886df7984cff3fca35ef8 (diff) |
Fixed Tensor alloc for some modules (For CUDA)
Diffstat (limited to 'CDivTable.lua')
-rw-r--r-- | CDivTable.lua | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/CDivTable.lua b/CDivTable.lua index f91d024..582b362 100644 --- a/CDivTable.lua +++ b/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 |