Compute sum_k M[k] @ V[k], the sum of the p matrix products, and return the (n, 1) result:
sum_k M[k] @ V[k]
p
(n, 1)
M
(p, n, n)
V
(p, n, 1)
This one's part of Premium. Unlock the full NumPy track plus every other premium problem on the site.
solve(...)