Mercurial > repos > public > sbplib_julia
diff src/LazyTensors/lazy_tensor_operations.jl @ 392:418cfd945715 feature/lazy_linear_map
Fix bug in range_size and domain_size for LazyLinearMap and expand the test
| author | Vidar Stiernström <vidar.stiernstrom@it.uu.se> |
|---|---|
| date | Fri, 02 Oct 2020 13:43:36 +0200 |
| parents | 8414c2334393 |
| children | b14eacf823b6 |
line wrap: on
line diff
--- a/src/LazyTensors/lazy_tensor_operations.jl Wed Sep 30 21:15:42 2020 +0200 +++ b/src/LazyTensors/lazy_tensor_operations.jl Fri Oct 02 13:43:36 2020 +0200 @@ -113,8 +113,8 @@ end export LazyLinearMap -range_size(llm::LazyLinearMap) = size(llm.A)[llm.range_indicies...] -domain_size(llm::LazyLinearMap) = size(llm.A)[llm.domain_indicies...] +range_size(llm::LazyLinearMap) = size(llm.A)[[llm.range_indicies...]] +domain_size(llm::LazyLinearMap) = size(llm.A)[[llm.domain_indicies...]] function apply(llm::LazyLinearMap{T,R,D}, v::AbstractArray{T,D}, I::Vararg{Index,R}) where {T,R,D} view_index = ntuple(i->:,ndims(llm.A)) @@ -126,4 +126,3 @@ return sum(A_view.*v) end -