import numpy as np
A = np.eye(20)
A @ A               # `@` is the matrix multiplication operator