pystencils.sympyextensions.math.multidimensional_sum

pystencils.sympyextensions.math.multidimensional_sum(i, dim)

Multidimensional summation

Example

>>> list(multidimensional_sum(2, dim=3))
[(0, 0), (0, 1), (0, 2), (1, 0), (1, 1), (1, 2), (2, 0), (2, 1), (2, 2)]