Commit 77152507 authored by Jakob Knollmüller's avatar Jakob Knollmüller
Browse files

more indices

parent 56fc35c0
Pipeline #94616 failed with stages
in 5 minutes and 25 seconds
......@@ -114,7 +114,7 @@ class DiagonalSelector(LinearOperator):
def apply(self, x, mode):
self._check_mode(mode)
if mode == self.TIMES:
result = np.diag(x.val[0])
result = np.diag(x.val)
return makeField(self._target,result)
return makeField(self._domain,np.diag(x.val).reshape(self._domain.shape))
......
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