Update linearchain.py (#113)

harvardnlp · Oct 13, 2021 · 5328ec5 · 5328ec5
1 parent a484559
commit 5328ec5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_struct/linearchain.py b/torch_struct/linearchain.py
@@ -53,7 +53,7 @@ def logpartition(self, log_potentials, lengths=None, force_grad=False):
         chart = self._chart((batch, bin_N, C, C), log_potentials, force_grad)
 
         # Init
-        init = torch.zeros(*chart.shape).bool()
+        init = torch.zeros_like(chart).bool()
         init.diagonal(0, 3, 4).fill_(True)
         chart = semiring.fill(chart, init, semiring.one)