np.identity(n) creates an identity matrix of size n x n where diagonal elements are 1, and the rest are 0.