Mercurial > repos > public > sbplib_julia

comparison src/LazyTensors/lazy_tensor_operations.jl @ 458:41f9cb6ee5a7 feature/inflated_tensormapping

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Moar tests and moar type stableness
author Jonatan Werpers <jonatan@werpers.com>
date Wed, 21 Oct 2020 20:57:38 +0200
parents 8fb6a5611c7a
children 11e58c49fb46 c364e2908c6e
comparison
equal deleted inserted replaced
457:8fb6a5611c7a 458:41f9cb6ee5a7
228 ``` 228 ```
229 (1,2,3,4) -> (1,:,:,4), (2,3) 229 (1,2,3,4) -> (1,:,:,4), (2,3)
230 ``` 230 ```
231 """ 231 """
232 function split_index(itm::InflatedTensorMapping{T,R,D}, I::Vararg{Any,R}) where {T,R,D} 232 function split_index(itm::InflatedTensorMapping{T,R,D}, I::Vararg{Any,R}) where {T,R,D}
233 I_before = I[1:range_dim(itm.before)] 233 I_before = slice_tuple(I, Val(1), Val(range_dim(itm.before)))
234 I_after = slice_tuple(I,Val(R-range_dim(itm.after)+1),Val(R)) 234 I_after = slice_tuple(I, Val(R-range_dim(itm.after)+1), Val(R))
235 235
236 view_index = (I_before..., ntuple((i)->:,domain_dim(itm.tm))..., I_after...) 236 view_index = (I_before..., ntuple((i)->:,domain_dim(itm.tm))..., I_after...)
237 inner_index = slice_tuple(I, Val(range_dim(itm.before)+1), Val(R-range_dim(itm.after))) 237 inner_index = slice_tuple(I, Val(range_dim(itm.before)+1), Val(R-range_dim(itm.after)))
238 238
239 return (view_index, inner_index) 239 return (view_index, inner_index)