From 0b3ce31a3e5d3140289cff32f4a7c83b8c44c42f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Bart=20van=20Merri=C3=ABnboer?=
 <bart.vanmerrienboer@gmail.com>
Date: Thu, 21 Jul 2016 13:18:32 -0400
Subject: [PATCH] Fix wrong zero gradient for assignment in direct mode (#139)

---
 src/gradfuns.lua | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/gradfuns.lua b/src/gradfuns.lua
index 80fade1..24306e7 100644
--- 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,
 }