diff options
author | soumith <soumith@fb.com> | 2015-07-28 20:41:02 +0300 |
---|---|---|
committer | soumith <soumith@fb.com> | 2015-07-28 20:41:02 +0300 |
commit | 841227a5dd30d407c254941dd36c5ec17bd83daf (patch) | |
tree | 923052f6a7a9c515a5b0399312cfb5a98eadb7a7 /CDivTable.lua | |
parent | 194522f1ba96432ab19c176e23a0b9b981174770 (diff) |
fixing table modules to return correct number of gradInputs
Diffstat (limited to 'CDivTable.lua')
-rw-r--r-- | CDivTable.lua | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/CDivTable.lua b/CDivTable.lua index 582b362..bf044c9 100644 --- a/CDivTable.lua +++ b/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 |