Skip to content

Commit 3997ab2

Browse files
authored
PlusOp in TensorSpace Multiplication (#263)
* PlusOp in TensorSpace Multiplication * clean up code
1 parent e7007c8 commit 3997ab2

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

src/PDE/KroneckerOperator.jl

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -402,10 +402,12 @@ Conversion(a::TensorSpace,b::TensorSpace) = ConversionWrapper(promote_type(prect
402402

403403

404404

405-
function Multiplication(f::Fun{TS},S::TensorSpace) where {TS<:TensorSpace}
406-
lr=LowRankFun(f)
407-
ops=map(kron,map(a->Multiplication(a,S.spaces[1]),lr.A),map(a->Multiplication(a,S.spaces[2]),lr.B))
408-
MultiplicationWrapper(f,+(ops...))
405+
function Multiplication(f::Fun{<:TensorSpace}, S::TensorSpace)
406+
lr = LowRankFun(f)
407+
MAs = map(a->Multiplication(a,S.spaces[1]), lr.A)
408+
MBs = map(a->Multiplication(a,S.spaces[2]), lr.B)
409+
ops = map(kron, MAs, MBs)
410+
MultiplicationWrapper(f, PlusOperator(ops))
409411
end
410412

411413
## Functionals

0 commit comments

Comments
 (0)