def mldivide(a, b): dimensions = a.shape if dimensions[0] == dimensions[1]: return la.solve(a, b) else: return la.lstsq(a, b)[0]