Fix wrong zero gradient for assignment in direct mode (#139)

twitter-archive · Jul 21, 2016 · 0b3ce31 · 0b3ce31
1 parent e13a48a
commit 0b3ce31
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/gradfuns.lua b/src/gradfuns.lua
@@ -152,7 +152,12 @@ functions.set = {
       return nil
    end,
    function(g, ans, x, k, v)
-      return g[k]
+      local gk = getValue(g[k])
+      if type(gk) == 'number' then
+        return gk
+      else
+        return torch.clone(gk)
+      end
    end,
 }