pymc.model.core.Model.profile

Model.profile(outs, *, n=1000, point=None, profile=True, **compile_fn_kwargs)

Compile and profile a PyTensor function which returns outs and takes values of model vars as a dict as an argument.

Parameters:

outsPyTensor variable or iterable of PyTensor variables

nint, default 1000

Number of iterations to run

pointPoint

Point to pass to the function

profileTrue or ProfileStats

compile_fn_kwargs

Compilation kwargs for pymc.model.core.Model.compile_fn()

Returns:

pytensor.compile.profiling.ProfileStats

Use .summary() to print stats.