Given p matrices M of shape (p, n, n) and p vectors V of shape
(p, n, 1), compute sum_k M[k] @ V[k] (the sum of the p matrix products) in a
single vectorized call rather than a Python loop. Return the (n, 1) result.
Sign in to write and run your own code.
solve(...)