vec = np.reshape(vec / np.linalg.norm(vec.ravel()), x0.shape)