def simplify(self, deep=False, **kwargs): if self.arg.is_Mul: coeff = [] args = [] for arg in self.arg.args: if arg.shape: args.append(arg) else: coeff.append(arg) if coeff: Mul = self.arg.func return abs(Mul(*coeff)) * self.func(Mul(*args)) return Function.simplify(self, deep=deep, **kwargs)