Mercurial > repos > public > sbplib_julia

comparison src/LazyTensors/lazy_tensor_operations.jl @ 538:848dec405332 feature/inflated_tensormapping_transpose

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Change variable names in apply and apply_transpose
author Jonatan Werpers <jonatan@werpers.com>
date Thu, 26 Nov 2020 17:51:17 +0100
parents 41e82a5d4d48
children 62d96e2cd165
comparison
equal deleted inserted replaced
537:41e82a5d4d48 538:848dec405332
257 domain_size(itm.after), 257 domain_size(itm.after),
258 ) 258 )
259 end 259 end
260 260
261 function apply(itm::InflatedTensorMapping{T,R,D}, v::AbstractArray{T,D}, I::Vararg{Any,R}) where {T,R,D} 261 function apply(itm::InflatedTensorMapping{T,R,D}, v::AbstractArray{T,D}, I::Vararg{Any,R}) where {T,R,D}
262 A = range_dim(itm.before) 262 dim_before = range_dim(itm.before)
263 B_domain = domain_dim(itm.tm) 263 dim_domain = domain_dim(itm.tm)
264 B_range = range_dim(itm.tm) 264 dim_range = range_dim(itm.tm)
265 C = range_dim(itm.after) 265 dim_after = range_dim(itm.after)
266 266
267 view_index, inner_index = split_index(Val(A), Val(B_domain), Val(B_range), Val(C), I...) 267 view_index, inner_index = split_index(Val(dim_before), Val(dim_domain), Val(dim_range), Val(dim_after), I...)
268 268
269 v_inner = view(v, view_index...) 269 v_inner = view(v, view_index...)
270 return apply(itm.tm, v_inner, inner_index...) 270 return apply(itm.tm, v_inner, inner_index...)
271 end 271 end
272 272
273 function apply_transpose(itm::InflatedTensorMapping{T,R,D}, v::AbstractArray{T,R}, I::Vararg{Any,D}) where {T,R,D} 273 function apply_transpose(itm::InflatedTensorMapping{T,R,D}, v::AbstractArray{T,R}, I::Vararg{Any,D}) where {T,R,D}
274 A = range_dim(itm.before) 274 dim_before = range_dim(itm.before)
275 B_domain = domain_dim(itm.tm) 275 dim_domain = domain_dim(itm.tm)
276 B_range = range_dim(itm.tm) 276 dim_range = range_dim(itm.tm)
277 C = range_dim(itm.after) 277 dim_after = range_dim(itm.after)
278 278
279 view_index, inner_index = split_index(Val(A), Val(B_range), Val(B_domain), Val(C), I...) 279 view_index, inner_index = split_index(Val(dim_before), Val(dim_range), Val(dim_domain), Val(dim_after), I...)
280 280
281 v_inner = view(v, view_index...) 281 v_inner = view(v, view_index...)
282 return apply_transpose(itm.tm, v_inner, inner_index...) 282 return apply_transpose(itm.tm, v_inner, inner_index...)
283 end 283 end
284 284