.

harvardnlp · Nov 13, 2019 · acae858 · acae858
1 parent ff623a0
commit acae858
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/torch_struct/semirings/fast_semirings.py b/torch_struct/semirings/fast_semirings.py
@@ -16,6 +16,10 @@ class FastLogSemiring(_BaseLog):
 
     Gradients give marginals.
     """
+    @staticmethod
+    def sum(xs, dim=-1):
+        return torch.logsumexp(xs, dim=dim)
+
 
     @staticmethod
     def matmul(a, b, dims=1):