Mercurial > repos > public > sbplib_julia
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 |