Multiply a (5, 5, 3) array A by a (5, 5) array B (broadcast B over the last axis). Return the result.
(5, 5, 3)
A
(5, 5)
B
solve(...)