diff src/LazyTensors/LazyTensors.jl @ 1300:a922aa69eb83 bugfix/lazy_tensors/tensorcomposition

Fix type inference of nested tensorcompositions by making the compositions right associative
author Vidar Stiernström <vidar.stiernstrom@it.uu.se>
date Wed, 26 Apr 2023 21:56:45 +0200
parents 93f87d5d9fbb
children dfb43fdac9fc 8b64df6cadba
line wrap: on
line diff
--- a/src/LazyTensors/LazyTensors.jl	Tue Feb 21 21:01:46 2023 +0100
+++ b/src/LazyTensors/LazyTensors.jl	Wed Apr 26 21:56:45 2023 +0200
@@ -30,6 +30,7 @@
 
 # Composing lazy tensors
 Base.:∘(s::LazyTensor, t::LazyTensor) = TensorComposition(s,t)
+Base.:∘(s::TensorComposition, t::LazyTensor) = s.t1∘(s.t2∘t)
 
 # Outer products of tensors
 ⊗(a::LazyTensor, b::LazyTensor) = LazyOuterProduct(a,b)