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):
res = x.ravel() if mode == self.TIMES else x.reshape(
self._domain.shape)
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()):
i0, i1 = self._size_array[ii:ii + 2]
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