Commit 9a2cc287 authored by Philipp Arras's avatar Philipp Arras

Support complex input as well

parent 83d9e58f
Pipeline #61946 passed with stages
in 9 minutes and 2 seconds
...@@ -57,7 +57,7 @@ class _DomRemover(LinearOperator): ...@@ -57,7 +57,7 @@ class _DomRemover(LinearOperator):
res = x.ravel() if mode == self.TIMES else x.reshape( res = x.ravel() if mode == self.TIMES else x.reshape(
self._domain.shape) self._domain.shape)
else: else:
res = np.empty(self.target.shape) if mode == self.TIMES else {} res = np.empty(self.target.shape, dtype=x.dtype) if mode == self.TIMES else {}
for ii, (kk, dd) in enumerate(self.domain.items()): for ii, (kk, dd) in enumerate(self.domain.items()):
i0, i1 = self._size_array[ii:ii + 2] i0, i1 = self._size_array[ii:ii + 2]
if mode == self.TIMES: if mode == self.TIMES:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment