Using a matrix as initial condition in scipy.integrate.solve_ivp

Question

I would like to pass a matrix as initial condition for scipy.integrate.solve_ivp.

With odeInt this was possible by flattening the matrix as: