Skip to content
Snippets Groups Projects
Commit 74831e32 authored by Jakob Roth's avatar Jakob Roth
Browse files

likelihood.py: add key order for einsum

parent b592d998
No related branches found
No related tags found
No related merge requests found
Pipeline #179428 passed
Stage: build_docker
Stage: testing
Hide whitespace changes
Inline Side-by-side
Showing
with 2 additions and 1 deletion
resolve/likelihood.py
+ 2
− 1
View file @ 74831e32
......@@ -47,7 +47,8 @@ class PolMatrixProduct(ift.Operator):
r3 = (rsh.ducktape(key3)).ducktape_left(key3)
einsum_dom = ift.MultiDomain.make({key1:ndom,
key2:ndom, key3:ndom})
einsum = ift.MultiLinearEinsum(einsum_dom, "ijr,jkr,klr->ilr")
einsum = ift.MultiLinearEinsum(einsum_dom, "ijr,jkr,klr->ilr",
key_order=(key1, key2, key3))
self._op = rsh.adjoint @ einsum @ (r1 + r2 + r3)
def apply(self, x):
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment