From d355ce6faf2c9601515fb582d753bbd0b689ea93 Mon Sep 17 00:00:00 2001
From: Ronan Collobert <ronan@collobert.com>
Date: Thu, 6 Dec 2012 12:31:40 +0100
Subject: fix Jacobian when using direct-updates

---
 Jacobian.lua | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

(limited to 'Jacobian.lua')

diff --git a/Jacobian.lua b/Jacobian.lua
index 16397a9..97237e8 100644
--- a/Jacobian.lua
+++ b/Jacobian.lua
@@ -40,10 +40,16 @@ function nn.Jacobian.backwardUpdate (module, input, param)
    local jacobian = torch.Tensor(param:nElement(),dout:nElement()):zero()
 
    -- original param
-   local origparam = param:clone()
+   local params = module:parameters()
+   local origparams = {}
+   for j=1,#params do
+      table.insert(origparams, params[j]:clone())
+   end
 
    for i=1,sdout:nElement() do
-      param:copy(origparam)
+      for j=1,#params do
+         params[j]:copy(origparams[j])
+      end
       dout:zero()
       sdout[i] = 1
       local din = module:updateGradInput(input, dout)
@@ -51,7 +57,9 @@ function nn.Jacobian.backwardUpdate (module, input, param)
       jacobian:select(2,i):copy(param)
    end
 
-   param:copy(origparam)
+   for j=1,#params do
+      params[j]:copy(origparams[j])
+   end
 
    return jacobian
 end
-- 
cgit v1.2.3