def cond(i, prev_elbo, elbo, qxt, qxt1):
return T.logical_and(i < max_iter, abs(prev_elbo - elbo) >= tol)
def cond(i, _, __, prev_elbo, curr_elbo):
with T.core.control_dependencies([T.core.print(curr_elbo)]):
prev_elbo = T.core.identity(prev_elbo)
return T.logical_and(
T.abs(curr_elbo - prev_elbo) > tol, i < max_iter)
def cond(i, elbo, prev_elbo, q):
return T.logical_and(i < max_iter, abs(elbo - prev_elbo) > tol)