Mercurial > repos > public > sbplib_julia
diff src/LazyTensors/lazy_tensor_operations.jl @ 1229:8f4259fbd39c refactor/LazyTensors/tuple_manipulation
Simplify split_index
| author | Jonatan Werpers <jonatan@werpers.com> |
|---|---|
| date | Sun, 19 Feb 2023 11:43:29 +0100 |
| parents | 6567e38b05ca |
| children | aa8579b7fc15 |
line wrap: on
line diff
--- a/src/LazyTensors/lazy_tensor_operations.jl Sun Feb 19 11:41:40 2023 +0100 +++ b/src/LazyTensors/lazy_tensor_operations.jl Sun Feb 19 11:43:29 2023 +0100 @@ -197,7 +197,7 @@ dim_range = range_dim(itm.tm) dim_after = range_dim(itm.after) - view_index, inner_index = split_index(Val(dim_before), Val(dim_domain), Val(dim_range), Val(dim_after), I...) + view_index, inner_index = split_index(dim_before, dim_domain, dim_range, dim_after, I...) v_inner = view(v, view_index...) return apply(itm.tm, v_inner, inner_index...) @@ -209,7 +209,7 @@ dim_range = range_dim(itm.tm) dim_after = range_dim(itm.after) - view_index, inner_index = split_index(Val(dim_before), Val(dim_range), Val(dim_domain), Val(dim_after), I...) + view_index, inner_index = split_index(dim_before, dim_range, dim_domain, dim_after, I...) v_inner = view(v, view_index...) return apply_transpose(itm.tm, v_inner, inner_index...)