comparison src/LazyTensors/lazy_tensor_operations.jl @ 525:7e6250c51eb2 feature/inflated_tensormapping_transpose

Fix bug in apply
author Jonatan Werpers <jonatan@werpers.com>
date Wed, 25 Nov 2020 15:24:55 +0100
parents fe86ac896377
children be152486d136
comparison
equal deleted inserted replaced
524:41c1760a7770 525:7e6250c51eb2
264 A = range_dim(itm.before) 264 A = range_dim(itm.before)
265 B_domain = domain_dim(itm.tm) 265 B_domain = domain_dim(itm.tm)
266 B_range = range_dim(itm.tm) 266 B_range = range_dim(itm.tm)
267 C = range_dim(itm.after) 267 C = range_dim(itm.after)
268 268
269 view_index, inner_index = split_index(Val(A), Val(B_range), Val(B_domain), Val(C), I...) 269 view_index, inner_index = split_index(Val(A), Val(B_domain), Val(B_range), Val(C), I...)
270 270
271 v_inner = view(v, view_index...) 271 v_inner = view(v, view_index...)
272 return apply(itm.tm, v_inner, inner_index...) 272 return apply(itm.tm, v_inner, inner_index...)
273 end 273 end
274 274