From efaa08cbf5fd56806a7c86e4eca1c985ff73e6d2 Mon Sep 17 00:00:00 2001
From: lucas_miranda <lucasmiranda42@gmail.com>
Date: Mon, 20 Jul 2020 22:34:32 +0200
Subject: [PATCH] test posterior with constant variance per component

---
 source/model_utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/source/model_utils.py b/source/model_utils.py
index 72b354ac..270c9098 100644
--- a/source/model_utils.py
+++ b/source/model_utils.py
@@ -286,7 +286,7 @@ class Entropy_regulariser(Layer):
     def call(self, z, **kwargs):
 
         entropy = K.sum(
-            tf.multiply(z, tf.where(~tf.math.is_inf(K.log(z)), K.log(z), 0)), axis=1
+            tf.multiply(z, tf.where(~tf.math.is_inf(K.log(z)), K.log(z), 0)), axis=0
         )
 
         # Adds metric that monitors dead neurons in the latent space
-- 
GitLab