pymc.model.core.Model.compile_fn

Model.compile_fn(outs, *, inputs=None, mode=None, point_fn=True, **kwargs)

Compiles a PyTensor function.

Parameters:

outsVariable or sequence of Variables

PyTensor variable or iterable of PyTensor variables.

inputssequence of Variables, optional

PyTensor input variables, defaults to pytensorf.inputvars(outs).

mode

PyTensor compilation mode, default=None.

point_fnbool

Whether to wrap the compiled function in a PointFunc, which takes a Point dictionary with model variable names and values as input.

Other keyword arguments

Any other keyword argument is sent to pymc.pytensorf.compile_pymc().

Returns:

Compiled PyTensor function