matrix = np.array([[[1., 0.], [3., 4.]], [[1., 0.], [2., 1.]]])