_jitcode.py 文件源码

def generate_jac_lambda(self,do_cse=False):
        """
        translates the symbolic Jacobain to a function using SymEngines `Lambdify` or `LambdifyCSE`. If the symbolic Jacobian has not been generated, it is generated by calling `generate_jac_sym`.

        Parameters
        ----------
        do_cse : boolean
            Whether a common-subexpression detection, namely `LambdifyCSE`, should be used.
        """

        self._prepare_lambdas()
        self._generate_jac_sym()

        jac_matrix = symengine.Matrix([
                [ordered_subs(entry,self._lambda_subs) for entry in line]
                for line in self.jac_sym
            ])

        lambdify = symengine.LambdifyCSE if do_cse else symengine.Lambdify
        core_jac = lambdify(self._lambda_args,jac_matrix)
        # self.jac = lambda t,Y,*c_pars: core_jac(np.hstack([t,Y,c_pars]))
        self.jac = _lambda_wrapper(core_jac,self.control_pars)