fix elbo full rank

pymc-devs · Sep 24, 2021 · 678568c · 678568c
1 parent 4eb12f3
commit 678568c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pymc3/variational/approximations.py b/pymc3/variational/approximations.py
@@ -175,7 +175,7 @@ def symbolic_logq_not_scaled(self):
         z0 = self.symbolic_initial
         diag = at.diagonal(self.L, 0, self.L.ndim - 2, self.L.ndim - 1)
         logdet = at.log(diag)
-        quaddist = ((z0) ** 2 + at.log(np.pi / 2.0)) / 2.0
+        quaddist = -0.5 * z0 ** 2 - at.log((2 * np.pi) ** 0.5)
         logq = quaddist - logdet
         return logq.sum(range(1, logq.ndim))