I = np.eye(np.shape(AA))