.

harvardnlp · Sep 8, 2019 · e979d19 · e979d19
1 parent 97d150e
commit e979d19
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/torch_struct/deptree.py b/torch_struct/deptree.py
@@ -62,8 +62,9 @@ def _dp(self, arc_scores, lengths=None, force_grad=False):
         batch, N, lengths = self._check_potentials(arc_scores, lengths)
 
         DIRS = 2
-        s = MySlice.apply
-
+        #s = MySlice.apply
+        def s(input, e, a, b, c, d):
+            return input[e, :, a:b, c:d]
         def stack(a, b):
             return torch.stack([a, b])