Mercurial > repos > public > sbplib_julia

comparison src/LazyTensors/lazy_tensor_operations.jl @ 1229:8f4259fbd39c refactor/LazyTensors/tuple_manipulation

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Simplify split_index
author Jonatan Werpers <jonatan@werpers.com>
date Sun, 19 Feb 2023 11:43:29 +0100
parents 6567e38b05ca
children aa8579b7fc15
comparison
equal deleted inserted replaced
1228:73f262a0a384 1229:8f4259fbd39c
195 dim_before = range_dim(itm.before) 195 dim_before = range_dim(itm.before)
196 dim_domain = domain_dim(itm.tm) 196 dim_domain = domain_dim(itm.tm)
197 dim_range = range_dim(itm.tm) 197 dim_range = range_dim(itm.tm)
198 dim_after = range_dim(itm.after) 198 dim_after = range_dim(itm.after)
199 199
200 view_index, inner_index = split_index(Val(dim_before), Val(dim_domain), Val(dim_range), Val(dim_after), I...) 200 view_index, inner_index = split_index(dim_before, dim_domain, dim_range, dim_after, I...)
201 201
202 v_inner = view(v, view_index...) 202 v_inner = view(v, view_index...)
203 return apply(itm.tm, v_inner, inner_index...) 203 return apply(itm.tm, v_inner, inner_index...)
204 end 204 end
205 205
207 dim_before = range_dim(itm.before) 207 dim_before = range_dim(itm.before)
208 dim_domain = domain_dim(itm.tm) 208 dim_domain = domain_dim(itm.tm)
209 dim_range = range_dim(itm.tm) 209 dim_range = range_dim(itm.tm)
210 dim_after = range_dim(itm.after) 210 dim_after = range_dim(itm.after)
211 211
212 view_index, inner_index = split_index(Val(dim_before), Val(dim_range), Val(dim_domain), Val(dim_after), I...) 212 view_index, inner_index = split_index(dim_before, dim_range, dim_domain, dim_after, I...)
213 213
214 v_inner = view(v, view_index...) 214 v_inner = view(v, view_index...)
215 return apply_transpose(itm.tm, v_inner, inner_index...) 215 return apply_transpose(itm.tm, v_inner, inner_index...)
216 end 216 end
217 217