Premium problem71. Multiply (5,5,3) by (5,5)

Hard Locked

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

Premium problem

This one's part of Premium. Unlock the full NumPy track plus every other premium problem on the site.

Implement solve(...)