Skip to content
Snippets Groups Projects

LinearEinsum: fix adjoint_times for pure transpose operations

Merged Lukas Platz requested to merge fixup_einsum_transpose into NIFTy_6
2 files
+ 22
2
Compare changes
  • Side-by-side
  • Inline
Files
2
@@ -262,8 +262,11 @@ class LinearEinsum(LinearOperator):
iss, oss, *_ = subscripts.split("->")
iss_spl = iss.split(",")
adj_iss = ",".join((",".join(iss_spl[:-1]), oss))
self._adj_sscr = "->".join((adj_iss, iss_spl[-1]))
if len(iss_spl) == 1:
self._adj_sscr = "->".join((oss, iss))
else:
adj_iss = ",".join((",".join(iss_spl[:-1]), oss))
self._adj_sscr = "->".join((adj_iss, iss_spl[-1]))
self._capability = self.TIMES | self.ADJOINT_TIMES
def apply(self, x, mode):
Loading