.

harvardnlp · Oct 4, 2019 · 9fb834e · 9fb834e
1 parent eb9dd9b
commit 9fb834e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_struct/linearchain.py b/torch_struct/linearchain.py
@@ -37,7 +37,7 @@ def _dp(self, edge, lengths=None, force_grad=False):
         semiring.one_(alpha[0].data)
         BATCH_DIM, N_DIM = 1, 2
 
-        for n in torch.arange(1, N):
+        for n in range(1, N):
             edge_store[n - 1][:] = semiring.times(
                 alpha[n - 1].view(ssize, batch, 1, C),
                 edge.index_select(N_DIM, n - 1).view(ssize, batch, C, C),