def log_q_W_global(self, z):
"""log_q_W samples over q for global vars
"""
mu = self.scale_grad(self.mean)
L = self.scale_grad(self.L)
z = z[self.global_slc]
return log_normal_mv(z, mu, chol=L, gpu_compat=self.gpu_compat)
def log_q_W_global(self, z):
"""
log_q_W samples over q for global vars
Gradient wrt mu, rho in density parametrization
is set to zero to lower variance of ELBO
"""
mu = self.scale_grad(self.mean)
L = self.scale_grad(self.L)
z = z[self.global_slc]
return log_normal_mv(z, mu, chol=L, gpu_compat=self.gpu_compat)