diff --git a/sympyextensions.py b/sympyextensions.py index e567d9cd85517c3b042d8d1cb83f45604838921d..489849ca7ebfe17a88e14cbf85c09ee31925f568 100644 --- a/sympyextensions.py +++ b/sympyextensions.py @@ -16,6 +16,10 @@ def allIn(a, b): return all(element in b for element in a) +def scalarProduct(a, b): + return sum(a_i * b_i for a_i, b_i in zip(a, b)) + + def normalizeProduct(product): """ Expects a sympy expression that can be interpreted as a product and