Есть ли какая-нибудь функция для суммирования по Матрице на Sympy?

Question

Когда я делаю суммирование по матрице:

import sympy
i = sy.symbols('i')
A=Matrix([1,i,3,3*i])
summation(A[i],(i,0,3))

Вот результаты:

IndexError: Invalid index a[i]

Я знаю, что могу решить эту проблему с помощью цикла.

Нет ли функции суммирования, поддерживающей матрицу Симпи?

Answer 1

Вы можете суммировать по вектору A a умножением на вектор единиц соответствующей формы.

A.T * ones(A.shape[0], 1)