Mercurial > repos > public > sbplib_julia

diff src/LazyTensors/lazy_tensor_operations.jl @ 414:f4c51b0350ba

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Merge feature/LazyLinearMap/transpose. Add apply_transpose method to LazyLinearMapping
author Jonatan Werpers <jonatan@werpers.com>
date Fri, 16 Oct 2020 09:34:06 +0200
parents 4aa59af074ef
children 814865d40f48 2808c41f5efa 46acb2560451
line wrap: on
line diff
--- a/src/LazyTensors/lazy_tensor_operations.jl	Thu Oct 15 21:06:28 2020 +0200
+++ b/src/LazyTensors/lazy_tensor_operations.jl	Fri Oct 16 09:34:06 2020 +0200
@@ -137,3 +137,7 @@
     A_view = @view llm.A[view_index...]
     return sum(A_view.*v)
 end
+
+function apply_transpose(llm::LazyLinearMap{T,R,D}, v::AbstractArray{T,R}, I::Vararg{Index,D}) where {T,R,D}
+    apply(LazyLinearMap(llm.A, llm.domain_indicies, llm.range_indicies), v, I...)
+end