leaves: Fix missing self object

braun-steven · Nov 2, 2023 · 1523ce8 · 1523ce8
1 parent dfed5db
commit 1523ce8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/simple_einet/layers/distributions/multivariate_normal.py b/simple_einet/layers/distributions/multivariate_normal.py
@@ -66,8 +66,8 @@ def scale_tril(self):
 
     def _get_base_distribution(self, ctx: SamplingContext = None, marginalized_scopes = None):
         # View means and scale_tril
-        means = self.means.view(self._num_dists, cardinality)
-        scale_tril = self.scale_tril.view(self._num_dists, cardinality, cardinality)
+        means = self.means.view(self._num_dists, self.cardinality)
+        scale_tril = self.scale_tril.view(self._num_dists, self.cardinality, self.cardinality)
 
 
         mv = CustomMultivariateNormalDist(