Mercurial > repos > public > sbplib_julia

comparison src/LazyTensors/lazy_tensor_operations.jl @ 1225:6567e38b05ca refactor/LazyTensors/tuple_manipulation

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Replace flatten_tuple with concatenate_tuple
author Jonatan Werpers <jonatan@werpers.com>
date Sat, 18 Feb 2023 12:24:44 +0100
parents 6f51160c7ca7
children 8f4259fbd39c
comparison
equal deleted inserted replaced
1224:e2f6dafb5d83 1225:6567e38b05ca
174 InflatedTensor(I1::IdentityTensor{T}, I2::IdentityTensor{T}) where T = InflatedTensor(I1,I2,IdentityTensor{T}()) 174 InflatedTensor(I1::IdentityTensor{T}, I2::IdentityTensor{T}) where T = InflatedTensor(I1,I2,IdentityTensor{T}())
175 175
176 # TODO: Implement some pretty printing in terms of ⊗. E.g InflatedTensor(I(3),B,I(2)) -> I(3)⊗B⊗I(2) 176 # TODO: Implement some pretty printing in terms of ⊗. E.g InflatedTensor(I(3),B,I(2)) -> I(3)⊗B⊗I(2)
177 177
178 function range_size(itm::InflatedTensor) 178 function range_size(itm::InflatedTensor)
179 return flatten_tuple( 179 return concatenate_tuples(
180 range_size(itm.before), 180 range_size(itm.before),
181 range_size(itm.tm), 181 range_size(itm.tm),
182 range_size(itm.after), 182 range_size(itm.after),
183 ) 183 )
184 end 184 end
185 185
186 function domain_size(itm::InflatedTensor) 186 function domain_size(itm::InflatedTensor)
187 return flatten_tuple( 187 return concatenate_tuples(
188 domain_size(itm.before), 188 domain_size(itm.before),
189 domain_size(itm.tm), 189 domain_size(itm.tm),
190 domain_size(itm.after), 190 domain_size(itm.after),
191 ) 191 )
192 end 192 end