implement LoopConcatenate._sum

evalf · Mar 11, 2024 · c96c6fa · c96c6fa
1 parent d7e8833
commit c96c6fa
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/nutils/evaluable.py b/nutils/evaluable.py
@@ -4529,6 +4529,12 @@ def _simplified(self):
             return
         return align(f, where, self.shape)
 
+    def _sum(self, axis):
+        if axis == self.ndim-1:
+            return loop_sum(Sum(self.func), self.index)
+        else:
+            return loop_concatenate(sum(self.func, axis), self.index)
+
     def _takediag(self, axis1, axis2):
         if axis1 < self.ndim-1 and axis2 < self.ndim-1:
             return Transpose.from_end(loop_concatenate(Transpose.to_end(_takediag(self.func, axis1, axis2), -2), self.index), -2)