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)
