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)