CMulTable updateGradInput now uses proper but slow gradient calculation. The old one is available in updateGradInput_efficient function.

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
+