fix sum dims

arrjon · arrjon · commit 8fbf7374ca95 · 2025-09-07T15:04:54.000+02:00
diff --git a/bayesflow/distributions/diagonal_normal.py b/bayesflow/distributions/diagonal_normal.py
@@ -91,7 +91,7 @@ def log_prob(self, samples: Tensor, *, normalize: bool = True) -> Tensor:
         result = -0.5 * ops.sum((samples - self._mean) ** 2 / self._std**2, axis=-1)
 
         if normalize:
-            log_normalization_constant = -0.5 * ops.sum(self.dims) * math.log(2.0 * math.pi) - ops.sum(
+            log_normalization_constant = -0.5 * np.sum(self.dims) * math.log(2.0 * math.pi) - ops.sum(
                 ops.log(self._std)
             )
             result += log_normalization_constant

Original file line number	Diff line number	Diff line change
`@@ -91,7 +91,7 @@ def log_prob(self, samples: Tensor, *, normalize: bool = True) -> Tensor:`
`91`	`91`	`result = -0.5 * ops.sum((samples - self._mean) 2 / self._std2, axis=-1)`
`92`	`92`
`93`	`93`	`if normalize:`
`94`		`- log_normalization_constant = -0.5 * ops.sum(self.dims) * math.log(2.0 * math.pi) - ops.sum(`
	`94`	`+ log_normalization_constant = -0.5 * np.sum(self.dims) * math.log(2.0 * math.pi) - ops.sum(`
`95`	`95`	`ops.log(self._std)`
`96`	`96`	`)`
`97`	`97`	`result += log_normalization_constant`