Commit 1e761a72 authored by Neel Shah's avatar Neel Shah
Browse files

fix functionality for applying to a multidimensional domain

parent b3ae70f0
Pipeline #106975 canceled with stages
......@@ -126,7 +126,14 @@ class MatrixProductOperator(EndomorphicOperator):
if self._spaces is None:
if not self._flatten:
res = m.dot(x.val)
if times:
res = np.tensordot(m,x.val,axes=len(x.shape))
else:
mat_axes = np.flip(self._mat_last_n)
field_axes = self._mat_first_n
res = np.tensordot(m,x.val,axes=(mat_axes,field_axes))
res = res.reshape(np.flip(res.shape))
else:
res = m.dot(x.val.flatten()).reshape(self._domain.shape)
return Field(self._domain, res)
......
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