PyMCModel.logp_dlogp_function

PyMCModel.logp_dlogp_function(grad_vars=None, tempered=False, **kwargs)

Compile an PyTensor function that computes logp and gradient.

Parameters:

• grad_vars (list of random variables, optional) – Compute the gradient with respect to those variables. If None, use all free random variables of this model.

• tempered (bool) – Compute the tempered logp free_logp + alpha * observed_logp. alpha can be changed using ValueGradFunction.set_weights([alpha]).