86. Sum of p Matrix Products

Hard

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.

Implement solve(...)