def std(self): return rho2sigma(self.rho)
def symbolic_logq_not_scaled(self): z0 = self.symbolic_initial std = rho2sigma(self.rho) logdet = at.log(std) logq = pm.Normal.dist().logp(z0) - logdet return logq.sum(range(1, logq.ndim))
def cov(self): var = rho2sigma(self.rho)**2 if self.batched: return batched_diag(var) else: return at.diag(var)
def __init__(self, rho=None, **kwargs): super().__init__(**kwargs) rho = self.add_param(rho, "rho") self.scale = rho2sigma(rho) self.shared_params = dict(rho=rho)