>>> np.identity(3) array([[1., 0., 0.], [0., 1., 0.], [0., 0., 1.]])
>>> np.eye(3) array([[1., 0., 0.], [0., 1., 0.], [0., 0., 1.]])