comparison src/LazyTensors/lazy_tensor_operations.jl @ 1014:67969bd7e642 feature/lazy_tensors/pretty_printing

Remove todo
author Jonatan Werpers <jonatan@werpers.com>
date Mon, 21 Mar 2022 09:32:26 +0100
parents 2c1a0722ddb9
children 1e8270c18edb
comparison
equal deleted inserted replaced
1012:2c1a0722ddb9 1014:67969bd7e642
169 169
170 InflatedLazyTensor(before::IdentityTensor, tm::LazyTensor{T}) where T = InflatedLazyTensor(before,tm,IdentityTensor{T}()) 170 InflatedLazyTensor(before::IdentityTensor, tm::LazyTensor{T}) where T = InflatedLazyTensor(before,tm,IdentityTensor{T}())
171 InflatedLazyTensor(tm::LazyTensor{T}, after::IdentityTensor) where T = InflatedLazyTensor(IdentityTensor{T}(),tm,after) 171 InflatedLazyTensor(tm::LazyTensor{T}, after::IdentityTensor) where T = InflatedLazyTensor(IdentityTensor{T}(),tm,after)
172 # Resolve ambiguity between the two previous methods 172 # Resolve ambiguity between the two previous methods
173 InflatedLazyTensor(I1::IdentityTensor{T}, I2::IdentityTensor{T}) where T = InflatedLazyTensor(I1,I2,IdentityTensor{T}()) 173 InflatedLazyTensor(I1::IdentityTensor{T}, I2::IdentityTensor{T}) where T = InflatedLazyTensor(I1,I2,IdentityTensor{T}())
174
175 # TODO: Implement some pretty printing in terms of ⊗. E.g InflatedLazyTensor(I(3),B,I(2)) -> I(3)⊗B⊗I(2)
176 174
177 function range_size(itm::InflatedLazyTensor) 175 function range_size(itm::InflatedLazyTensor)
178 return flatten_tuple( 176 return flatten_tuple(
179 range_size(itm.before), 177 range_size(itm.before),
180 range_size(itm.tm), 178 range_size(itm.tm),