3c38ff6794c12b84059b7a442e95b884c4f86a16,mcex/core.py,,model_logp_dlogp,#Any#Any#Any#,85

Before Change

    
    mapping = IASpaceMap(dvars)
    
    logp = logp_calc(model)
    calculations = [logp] + [grad(logp, var) for var in dvars]
        
    f = function(model.vars, calculations, allow_input_downcast = True, mode = mode)
    def fn(state):

After Change

    return model_func(model, dlogp_calc(model, dvars), mode)
    
def model_logp_dlogp(model, dvars = None, mode = None ):    
    return model_func(model, [logp_calc(model), dlogp_calc(model, dvars)], mode)



The functions build graphs from other graphs

Instances