.

harvardnlp · Oct 31, 2019 · 1e89b98 · 1e89b98
1 parent f806c0e
commit 1e89b98
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/torch_struct/semirings.py b/torch_struct/semirings.py
@@ -136,11 +136,11 @@ def back(x):
 
 def unaccumulate_(a, b, grad_output, fn, step=10000):
     slices = []
-    # a_grad = a.clone().fill_(0)    
-    # b_grad = b.clone().fill_(0)
+    a_grad = a.clone().fill_(0)    
+    b_grad = b.clone().fill_(0)
     # print("chcek", a_grad.shape)    
-    a_grad = torch.tensor(0.0, device=a.device).set_(a.clone().storage(), a.storage_offset(), a.size(), a.stride()).fill_(0)
-    b_grad = torch.tensor(0.0, device=b.device).set_(b.clone().storage(), b.storage_offset(), b.size(), b.stride()).fill_(0)
+    # a_grad = torch.tensor(0.0, device=a.device).set_(a.clone().storage(), a.storage_offset(), a.size(), a.stride()).fill_(0)
+    # b_grad = torch.tensor(0.0, device=b.device).set_(b.clone().storage(), b.storage_offset(), b.size(), b.stride()).fill_(0)
 
     total = 1
     for s in grad_output.shape: