Exemplo n.º 1
0
 def __call__(self, model, guide, *args, **kwargs):
     if self.options.get("optimize", None):
         with funsor.interpreter.interpretation(funsor.optimizer.optimize):
             elbo_expr = elbo(model, guide, *args, **kwargs)
         return funsor.reinterpret(elbo_expr)
     return elbo(model, guide, *args, **kwargs)
Exemplo n.º 2
0
def test_reinterpret(expr):
    x = eval(expr)
    assert funsor.reinterpret(x) is x