Numpy insert matrix values with matrix index

Question

I have the following code which creates a 4D grid matrix and I am looking to insert the rolled 2D vals matrix into this grid.

import num