.

harvardnlp · Nov 26, 2019 · 9e9a863 · 9e9a863
1 parent e92deaa
commit 9e9a863
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_struct/linearchain.py b/torch_struct/linearchain.py
@@ -76,7 +76,7 @@ def _dp_scan(self, log_potentials, lengths=None, force_grad=False):
         mask = torch.arange(bin_N).view(1, bin_N).expand(batch, bin_N)
         mask = mask >= (lengths - 1).view(batch, 1)
         lp.data[:, mask.view(-1)] = semiring.zero
-        c.data[:, (1- mask).view(-1)] = semiring.zero
+        c.data[:, (~mask).view(-1)] = semiring.zero
         c[:] = semiring.sum(torch.stack([c, lp], dim=-1))