Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/torch/nn.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorIvo Danihelka <ivo@danihelka.net>2014-10-29 18:54:03 +0300
committerIvo Danihelka <ivo@danihelka.net>2014-10-29 18:54:03 +0300
commit3e11d6339ebe442c857d3cb0cf76c1880596b64f (patch)
tree659aec176506dd02f2f0990ade36db61f8cd735b /CSubTable.lua
parent00b507509a9f2b73cea35b2fc6fc38632a225257 (diff)
Updated CSubTable to have gradInput[2]:size() equal to input[2]:size().
Diffstat (limited to 'CSubTable.lua')
-rw-r--r--CSubTable.lua2
1 files changed, 1 insertions, 1 deletions
diff --git a/CSubTable.lua b/CSubTable.lua
index 060dee5..b929e48 100644
--- a/CSubTable.lua
+++ b/CSubTable.lua
@@ -16,6 +16,6 @@ function CSubTable:updateGradInput(input, gradOutput)
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)
+ self.gradInput[2]:resizeAs(input[2]):copy(gradOutput):mul(-1)
return self.gradInput
end