Mercurial > repos > public > sbplib_julia
diff src/LazyTensors/lazy_tensor_operations.jl @ 538:848dec405332 feature/inflated_tensormapping_transpose
Change variable names in apply and apply_transpose
| author | Jonatan Werpers <jonatan@werpers.com> |
|---|---|
| date | Thu, 26 Nov 2020 17:51:17 +0100 |
| parents | 41e82a5d4d48 |
| children | 62d96e2cd165 |
line wrap: on
line diff
--- a/src/LazyTensors/lazy_tensor_operations.jl Thu Nov 26 17:43:05 2020 +0100 +++ b/src/LazyTensors/lazy_tensor_operations.jl Thu Nov 26 17:51:17 2020 +0100 @@ -259,24 +259,24 @@ end function apply(itm::InflatedTensorMapping{T,R,D}, v::AbstractArray{T,D}, I::Vararg{Any,R}) 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) + dim_before = range_dim(itm.before) + dim_domain = domain_dim(itm.tm) + dim_range = range_dim(itm.tm) + dim_after = range_dim(itm.after) - view_index, inner_index = split_index(Val(A), Val(B_domain), Val(B_range), Val(C), I...) + view_index, inner_index = split_index(Val(dim_before), Val(dim_domain), Val(dim_range), Val(dim_after), I...) v_inner = view(v, view_index...) 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) + dim_before = range_dim(itm.before) + dim_domain = domain_dim(itm.tm) + dim_range = range_dim(itm.tm) + dim_after = range_dim(itm.after) - view_index, inner_index = split_index(Val(A), Val(B_range), Val(B_domain), Val(C), I...) + view_index, inner_index = split_index(Val(dim_before), Val(dim_range), Val(dim_domain), Val(dim_after), I...) v_inner = view(v, view_index...) return apply_transpose(itm.tm, v_inner, inner_index...)