Merge pull request #16 from AaltoML/develop

Sparse EP with multi-dimensional x
AaltoML · Dec 19, 2022 · 77a9a53 · 77a9a53
2 parents 02eecea + cdd257d
commit 77a9a53
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bayesnewton/ops.py b/bayesnewton/ops.py
@@ -149,7 +149,7 @@ def sparse_conditional_post_to_data(kernel, post_mean, post_cov, X, Z):
         mean: posterior mean [N, 1]
         covariance: posterior covariance [N, N]
     """
-    X = X.reshape(X.shape[0], -1)
+    X = X.reshape(-1, Z.shape[1])
     Kff = kernel(X, X)
     Kuf = kernel(Z, X)
     Kuu = kernel(Z, Z)