How can I apply Scipy gmres to a 3D array without iterating through the array?

Question

So gmres solves for x with a system of equations in the form Ax = b. In this case my A matrix has shape 4x4x100, b has a shape 4x1x100 and I\'m expecting an output x with shape