changeset 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 a9ac86f6be8a
children e4dc576553cb
files src/LazyTensors/LazyTensors.jl
diffstat 1 files changed, 1 insertions(+), 0 deletions(-) [+]
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)