From 5d90b7385cb0a2ccdd25f9af89a81049ecefa7ca Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jakob=20Knollm=C3=BCller?= <jakobknollmuller@MBP-ORIGINS-Jakob.fritz.box> Date: Mon, 1 Mar 2021 11:48:10 +0100 Subject: [PATCH] fix --- src/library/variational_models.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/library/variational_models.py b/src/library/variational_models.py index ac405aba9..7931f49a8 100644 --- a/src/library/variational_models.py +++ b/src/library/variational_models.py @@ -55,8 +55,7 @@ class FullCovarianceModel(): co_part = PartialExtractor(matmul_setup_dom, co.domain) lat_part = PartialExtractor(matmul_setup_dom, lat.domain) matmul_setup = lat_part.adjoint @ lat.adjoint @ lat + co_part.adjoint @ co.adjoint @ cov - breakpoint() - MatMult = MultiLinearEinsum(matmul_setup.domain,'ij,ki->jk', key_order=('co','latent')) + MatMult = MultiLinearEinsum(matmul_setup.target,'ij,ki->jk', key_order=('co','latent')) Resp = Respacer(MatMult.target, mean.target) self.generator = self.Flat.adjoint @ (mean + Resp @ MatMult @ matmul_setup) -- GitLab