diff options
author | Clement Farabet <clement.farabet@gmail.com> | 2013-03-22 20:08:10 +0400 |
---|---|---|
committer | Clement Farabet <clement.farabet@gmail.com> | 2013-03-22 20:08:10 +0400 |
commit | e5d4dc79e482edb08e980189daeb62f9cd9fc86e (patch) | |
tree | bb636feb29f83c6755f9dcba815522b7bb4c28fb | |
parent | ee2834160ad08e3ff63d807cfbf5e957bff1caa1 (diff) | |
parent | 1e6fe1c4194d16c8245642f46454ce27c9cec48a (diff) |
Merge branch 'master' of github.com:andresy/torch
-rw-r--r-- | CMulTable.lua | 16 |
1 files changed, 15 insertions, 1 deletions
diff --git a/CMulTable.lua b/CMulTable.lua index f82776b..0e327be 100644 --- a/CMulTable.lua +++ b/CMulTable.lua @@ -14,7 +14,7 @@ function CMulTable:updateOutput(input) return self.output end -function CMulTable:updateGradInput(input, gradOutput) +function CMulTable:updateGradInput_efficient(input, gradOutput) self.tout = self.tout or input[1].new() self.tout:resizeAs(self.output) for i=1,#input do @@ -25,3 +25,17 @@ function CMulTable:updateGradInput(input, gradOutput) end return self.gradInput end + +function CMulTable:updateGradInput(input, gradOutput) + for i=1,#input do + self.gradInput[i] = self.gradInput[i] or input[1].new() + self.gradInput[i]:resizeAs(input[i]):copy(gradOutput) + for j=1,#input do + if i~=j then + self.gradInput[i]:cmul(input[j]) + end + end + end + return self.gradInput +end + |