Mercurial > repos > public > sbplib_julia
diff src/LazyTensors/lazy_tensor_operations.jl @ 526:be152486d136 feature/inflated_tensormapping_transpose
Implement apply_transpose with tests
| author | Jonatan Werpers <jonatan@werpers.com> |
|---|---|
| date | Wed, 25 Nov 2020 15:59:04 +0100 |
| parents | 7e6250c51eb2 |
| children | f9bc746f37da |
line wrap: on
line diff
--- a/src/LazyTensors/lazy_tensor_operations.jl Wed Nov 25 15:24:55 2020 +0100 +++ b/src/LazyTensors/lazy_tensor_operations.jl Wed Nov 25 15:59:04 2020 +0100 @@ -272,6 +272,18 @@ return apply(itm.tm, v_inner, inner_index...) end +function apply_transpose(itm::InflatedTensorMapping{T,R,D}, v::AbstractArray{T,R}, I::Vararg{Any,D}) where {T,R,D} + A = range_dim(itm.before) + B_domain = domain_dim(itm.tm) + B_range = range_dim(itm.tm) + C = range_dim(itm.after) + + view_index, inner_index = split_index(Val(A), Val(B_range), Val(B_domain), Val(C), I...) + + v_inner = view(v, view_index...) + return apply_transpose(itm.tm, v_inner, inner_index...) +end + """ split_index(:Val{A}, ::Val{B_view}, ::Val{B_middle}, ::Val{C}, I...)