sbplib_julia: src/LazyTensors/lazy_tensor

annotate src/LazyTensors/lazy_tensor_operations.jl @ 971:bc12be1b1ae5 feature/variable_derivatives

Fix typo

author	Jonatan Werpers <jonatan@werpers.com>
date	Mon, 14 Mar 2022 10:20:39 +0100
parents	e79debd10f7d
children	df562695b1b5

rev	line source
959 e9752c1e92f8 Fix conflict from merge Jonatan Werpers <jonatan@werpers.com> parents: 958 diff changeset	1 using Sbplib.RegionIndices
e9752c1e92f8 Fix conflict from merge Jonatan Werpers <jonatan@werpers.com> parents: 958 diff changeset	2
944 4a4ef4bf6cb9 Move exports to the top of the files Jonatan Werpers <jonatan@werpers.com> parents: 943 diff changeset	3 export LazyTensorMappingApplication
4a4ef4bf6cb9 Move exports to the top of the files Jonatan Werpers <jonatan@werpers.com> parents: 943 diff changeset	4 export LazyTensorMappingTranspose
4a4ef4bf6cb9 Move exports to the top of the files Jonatan Werpers <jonatan@werpers.com> parents: 943 diff changeset	5 export TensorMappingComposition
4a4ef4bf6cb9 Move exports to the top of the files Jonatan Werpers <jonatan@werpers.com> parents: 943 diff changeset	6 export LazyLinearMap
4a4ef4bf6cb9 Move exports to the top of the files Jonatan Werpers <jonatan@werpers.com> parents: 943 diff changeset	7 export IdentityMapping
4a4ef4bf6cb9 Move exports to the top of the files Jonatan Werpers <jonatan@werpers.com> parents: 943 diff changeset	8 export InflatedTensorMapping
4a4ef4bf6cb9 Move exports to the top of the files Jonatan Werpers <jonatan@werpers.com> parents: 943 diff changeset	9 export LazyOuterProduct
4a4ef4bf6cb9 Move exports to the top of the files Jonatan Werpers <jonatan@werpers.com> parents: 943 diff changeset	10 export ⊗
4a4ef4bf6cb9 Move exports to the top of the files Jonatan Werpers <jonatan@werpers.com> parents: 943 diff changeset	11 export SizeMismatch
4a4ef4bf6cb9 Move exports to the top of the files Jonatan Werpers <jonatan@werpers.com> parents: 943 diff changeset	12
196 b3c252280a19 Move LazyArray and make LazyTensorMappingApplication and LazyElementwiseOperation subtypes of it Jonatan Werpers <jonatan@werpers.com> parents: 195 diff changeset	13 """
b3c252280a19 Move LazyArray and make LazyTensorMappingApplication and LazyElementwiseOperation subtypes of it Jonatan Werpers <jonatan@werpers.com> parents: 195 diff changeset	14 LazyTensorMappingApplication{T,R,D} <: LazyArray{T,R}
190 8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	15
8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	16 Struct for lazy application of a TensorMapping. Created using `*`.
8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	17
8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	18 Allows the result of a `TensorMapping` applied to a vector to be treated as an `AbstractArray`.
8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	19 With a mapping `m` and a vector `v` the LazyTensorMappingApplication object can be created by `m*v`.
8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	20 The actual result will be calcualted when indexing into `m*v`.
8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	21 """
942 7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication Jonatan Werpers <jonatan@werpers.com> parents: 838 diff changeset	22 struct LazyTensorMappingApplication{T,R,D, TM<:TensorMapping{<:Any,R,D}, AA<:AbstractArray{<:Any,D}} <: LazyArray{T,R}
378 946516954c85 Use concrete types in LazyTensorApplication Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	23 t::TM
946516954c85 Use concrete types in LazyTensorApplication Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	24 o::AA
942 7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication Jonatan Werpers <jonatan@werpers.com> parents: 838 diff changeset	25
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication Jonatan Werpers <jonatan@werpers.com> parents: 838 diff changeset	26 function LazyTensorMappingApplication(t::TensorMapping{<:Any,R,D}, o::AbstractArray{<:Any,D}) where {R,D}
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication Jonatan Werpers <jonatan@werpers.com> parents: 838 diff changeset	27 T = promote_type(eltype(t), eltype(o))
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication Jonatan Werpers <jonatan@werpers.com> parents: 838 diff changeset	28 return new{T,R,D,typeof(t), typeof(o)}(t,o)
7829c09f8137 Add promotion calculation of element type for LazyTensorMappingApplication Jonatan Werpers <jonatan@werpers.com> parents: 838 diff changeset	29 end
190 8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	30 end
348 7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 333 diff changeset	31 # TODO: Do boundschecking on creation!
190 8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	32
547 53828d3ed132 Remove unnecessary type parameters Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 541 diff changeset	33 Base.getindex(ta::LazyTensorMappingApplication{T,R}, I::Vararg{Any,R}) where {T,R} = apply(ta.t, ta.o, I...)
953 6a2e30fa3bd8 Fix typo Jonatan Werpers <jonatan@werpers.com> parents: 941 diff changeset	34 Base.getindex(ta::LazyTensorMappingApplication{T,1}, I::CartesianIndex{1}) where {T} = apply(ta.t, ta.o, I.I...) # Would otherwise be caught in the previous method.
359 ba46a952a450 Remove unnecessary type parameters Jonatan Werpers <jonatan@werpers.com> parents: 348 diff changeset	35 Base.size(ta::LazyTensorMappingApplication) = range_size(ta.t)
190 8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	36 # TODO: What else is needed to implement the AbstractArray interface?
8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	37
421 2808c41f5efa Change defs of * for TensorApplication to fix bug Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	38 Base.:*(a::TensorMapping, v::AbstractArray) = LazyTensorMappingApplication(a,v)
2808c41f5efa Change defs of * for TensorApplication to fix bug Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	39 Base.:(a::TensorMapping, b::TensorMapping) = throw(MethodError(Base.:,(a,b)))
2808c41f5efa Change defs of * for TensorApplication to fix bug Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	40 Base.:(a::TensorMapping, args::Union{TensorMapping, AbstractArray}...) = foldr(,(a,args...))
2808c41f5efa Change defs of * for TensorApplication to fix bug Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	41
190 8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	42 # # We need the associativity to be a→b→c = a→(b→c), which is the case for '→'
274 11010bb74260 Dispatch getindex for TensorMappingApplication on region indices. Dispatch apply Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 267 diff changeset	43 # # Should we overload some other infix binary opesrator?
190 8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	44 # →(tm::TensorMapping{T,R,D}, o::AbstractArray{T,D}) where {T,R,D} = LazyTensorMappingApplication(tm,o)
8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	45 # TODO: We need to be really careful about good error messages.
8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	46 # For example what happens if you try to multiply LazyTensorMappingApplication with a TensorMapping(wrong order)?
8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	47
194 d30d42a11566 Move all Lazy operation into the same module Jonatan Werpers <jonatan@werpers.com> parents: 190 diff changeset	48 """
197 a340fa91b1fc Move transpose def to the bottom of the file Jonatan Werpers <jonatan@werpers.com> parents: 196 diff changeset	49 LazyTensorMappingTranspose{T,R,D} <: TensorMapping{T,D,R}
a340fa91b1fc Move transpose def to the bottom of the file Jonatan Werpers <jonatan@werpers.com> parents: 196 diff changeset	50
a340fa91b1fc Move transpose def to the bottom of the file Jonatan Werpers <jonatan@werpers.com> parents: 196 diff changeset	51 Struct for lazy transpose of a TensorMapping.
a340fa91b1fc Move transpose def to the bottom of the file Jonatan Werpers <jonatan@werpers.com> parents: 196 diff changeset	52
a340fa91b1fc Move transpose def to the bottom of the file Jonatan Werpers <jonatan@werpers.com> parents: 196 diff changeset	53 If a mapping implements the the `apply_transpose` method this allows working with
a340fa91b1fc Move transpose def to the bottom of the file Jonatan Werpers <jonatan@werpers.com> parents: 196 diff changeset	54 the transpose of mapping `m` by using `m'`. `m'` will work as a regular TensorMapping lazily calling
a340fa91b1fc Move transpose def to the bottom of the file Jonatan Werpers <jonatan@werpers.com> parents: 196 diff changeset	55 the appropriate methods of `m`.
a340fa91b1fc Move transpose def to the bottom of the file Jonatan Werpers <jonatan@werpers.com> parents: 196 diff changeset	56 """
438 907b0510699f Change LazyTensorTranspose to have concrete fields Jonatan Werpers <jonatan@werpers.com> parents: 436 diff changeset	57 struct LazyTensorMappingTranspose{T,R,D, TM<:TensorMapping{T,R,D}} <: TensorMapping{T,D,R}
907b0510699f Change LazyTensorTranspose to have concrete fields Jonatan Werpers <jonatan@werpers.com> parents: 436 diff changeset	58 tm::TM
197 a340fa91b1fc Move transpose def to the bottom of the file Jonatan Werpers <jonatan@werpers.com> parents: 196 diff changeset	59 end
a340fa91b1fc Move transpose def to the bottom of the file Jonatan Werpers <jonatan@werpers.com> parents: 196 diff changeset	60
a340fa91b1fc Move transpose def to the bottom of the file Jonatan Werpers <jonatan@werpers.com> parents: 196 diff changeset	61 # # TBD: Should this be implemented on a type by type basis or through a trait to provide earlier errors?
348 7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 333 diff changeset	62 # Jonatan 2020-09-25: Is the problem that you can take the transpose of any TensorMapping even if it doesn't implement `apply_transpose`?
274 11010bb74260 Dispatch getindex for TensorMappingApplication on region indices. Dispatch apply Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 267 diff changeset	63 Base.adjoint(tm::TensorMapping) = LazyTensorMappingTranspose(tm)
11010bb74260 Dispatch getindex for TensorMappingApplication on region indices. Dispatch apply Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 267 diff changeset	64 Base.adjoint(tmt::LazyTensorMappingTranspose) = tmt.tm
197 a340fa91b1fc Move transpose def to the bottom of the file Jonatan Werpers <jonatan@werpers.com> parents: 196 diff changeset	65
943 fb060e98ac0a Remove more type assertions Jonatan Werpers <jonatan@werpers.com> parents: 942 diff changeset	66 apply(tmt::LazyTensorMappingTranspose{T,R,D}, v::AbstractArray{<:Any,R}, I::Vararg{Any,D}) where {T,R,D} = apply_transpose(tmt.tm, v, I...)
fb060e98ac0a Remove more type assertions Jonatan Werpers <jonatan@werpers.com> parents: 942 diff changeset	67 apply_transpose(tmt::LazyTensorMappingTranspose{T,R,D}, v::AbstractArray{<:Any,D}, I::Vararg{Any,R}) where {T,R,D} = apply(tmt.tm, v, I...)
197 a340fa91b1fc Move transpose def to the bottom of the file Jonatan Werpers <jonatan@werpers.com> parents: 196 diff changeset	68
359 ba46a952a450 Remove unnecessary type parameters Jonatan Werpers <jonatan@werpers.com> parents: 348 diff changeset	69 range_size(tmt::LazyTensorMappingTranspose) = domain_size(tmt.tm)
ba46a952a450 Remove unnecessary type parameters Jonatan Werpers <jonatan@werpers.com> parents: 348 diff changeset	70 domain_size(tmt::LazyTensorMappingTranspose) = range_size(tmt.tm)
197 a340fa91b1fc Move transpose def to the bottom of the file Jonatan Werpers <jonatan@werpers.com> parents: 196 diff changeset	71
210 2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 209 diff changeset	72
2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 209 diff changeset	73 struct LazyTensorMappingBinaryOperation{Op,T,R,D,T1<:TensorMapping{T,R,D},T2<:TensorMapping{T,R,D}} <: TensorMapping{T,D,R}
274 11010bb74260 Dispatch getindex for TensorMappingApplication on region indices. Dispatch apply Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 267 diff changeset	74 tm1::T1
11010bb74260 Dispatch getindex for TensorMappingApplication on region indices. Dispatch apply Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 267 diff changeset	75 tm2::T2
210 2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 209 diff changeset	76
274 11010bb74260 Dispatch getindex for TensorMappingApplication on region indices. Dispatch apply Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 267 diff changeset	77 @inline function LazyTensorMappingBinaryOperation{Op,T,R,D}(tm1::T1,tm2::T2) where {Op,T,R,D, T1<:TensorMapping{T,R,D},T2<:TensorMapping{T,R,D}}
11010bb74260 Dispatch getindex for TensorMappingApplication on region indices. Dispatch apply Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 267 diff changeset	78 return new{Op,T,R,D,T1,T2}(tm1,tm2)
210 2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 209 diff changeset	79 end
2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 209 diff changeset	80 end
348 7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 333 diff changeset	81 # TODO: Boundschecking in constructor.
210 2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 209 diff changeset	82
943 fb060e98ac0a Remove more type assertions Jonatan Werpers <jonatan@werpers.com> parents: 942 diff changeset	83 apply(tmBinOp::LazyTensorMappingBinaryOperation{:+,T,R,D}, v::AbstractArray{<:Any,D}, I::Vararg{Any,R}) where {T,R,D} = apply(tmBinOp.tm1, v, I...) + apply(tmBinOp.tm2, v, I...)
fb060e98ac0a Remove more type assertions Jonatan Werpers <jonatan@werpers.com> parents: 942 diff changeset	84 apply(tmBinOp::LazyTensorMappingBinaryOperation{:-,T,R,D}, v::AbstractArray{<:Any,D}, I::Vararg{Any,R}) where {T,R,D} = apply(tmBinOp.tm1, v, I...) - apply(tmBinOp.tm2, v, I...)
210 2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 209 diff changeset	85
547 53828d3ed132 Remove unnecessary type parameters Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 541 diff changeset	86 range_size(tmBinOp::LazyTensorMappingBinaryOperation) = range_size(tmBinOp.tm1)
53828d3ed132 Remove unnecessary type parameters Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 541 diff changeset	87 domain_size(tmBinOp::LazyTensorMappingBinaryOperation) = domain_size(tmBinOp.tm1)
210 2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 209 diff changeset	88
274 11010bb74260 Dispatch getindex for TensorMappingApplication on region indices. Dispatch apply Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 267 diff changeset	89 Base.:+(tm1::TensorMapping{T,R,D}, tm2::TensorMapping{T,R,D}) where {T,R,D} = LazyTensorMappingBinaryOperation{:+,T,R,D}(tm1,tm2)
11010bb74260 Dispatch getindex for TensorMappingApplication on region indices. Dispatch apply Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 267 diff changeset	90 Base.:-(tm1::TensorMapping{T,R,D}, tm2::TensorMapping{T,R,D}) where {T,R,D} = LazyTensorMappingBinaryOperation{:-,T,R,D}(tm1,tm2)
210 2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 209 diff changeset	91
412 d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 402 diff changeset	92 """
d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 402 diff changeset	93 TensorMappingComposition{T,R,K,D}
210 2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 209 diff changeset	94
838 76e5682d0e52 Fix a bunch of docstring mistakes Jonatan Werpers <jonatan@werpers.com> parents: 625 diff changeset	95 Lazily compose two `TensorMapping`s, so that they can be handled as a single `TensorMapping`.
412 d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 402 diff changeset	96 """
d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 402 diff changeset	97 struct TensorMappingComposition{T,R,K,D, TM1<:TensorMapping{T,R,K}, TM2<:TensorMapping{T,K,D}} <: TensorMapping{T,R,D}
d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 402 diff changeset	98 t1::TM1
d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 402 diff changeset	99 t2::TM2
190 8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	100
417 4c6604b7d990 Add dimension checking in the constructor Jonatan Werpers <jonatan@werpers.com> parents: 415 diff changeset	101 @inline function TensorMappingComposition(t1::TensorMapping{T,R,K}, t2::TensorMapping{T,K,D}) where {T,R,K,D}
485 4b49f03bdb98 Switch from DimensionMismatch to SizeMismatch for boundschecks on compositions Jonatan Werpers <jonatan@werpers.com> parents: 484 diff changeset	102 @boundscheck check_domain_size(t1, range_size(t2))
417 4c6604b7d990 Add dimension checking in the constructor Jonatan Werpers <jonatan@werpers.com> parents: 415 diff changeset	103 return new{T,R,K,D, typeof(t1), typeof(t2)}(t1,t2)
4c6604b7d990 Add dimension checking in the constructor Jonatan Werpers <jonatan@werpers.com> parents: 415 diff changeset	104 end
412 d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 402 diff changeset	105 end
190 8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	106
412 d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 402 diff changeset	107 range_size(tm::TensorMappingComposition) = range_size(tm.t1)
d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 402 diff changeset	108 domain_size(tm::TensorMappingComposition) = domain_size(tm.t2)
190 8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	109
943 fb060e98ac0a Remove more type assertions Jonatan Werpers <jonatan@werpers.com> parents: 942 diff changeset	110 function apply(c::TensorMappingComposition{T,R,K,D}, v::AbstractArray{<:Any,D}, I::Vararg{Any,R}) where {T,R,K,D}
418 264af2bb646f Switch to using * instead of calling the constructor Jonatan Werpers <jonatan@werpers.com> parents: 417 diff changeset	111 apply(c.t1, c.t2*v, I...)
412 d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 402 diff changeset	112 end
190 8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	113
943 fb060e98ac0a Remove more type assertions Jonatan Werpers <jonatan@werpers.com> parents: 942 diff changeset	114 function apply_transpose(c::TensorMappingComposition{T,R,K,D}, v::AbstractArray{<:Any,R}, I::Vararg{Any,D}) where {T,R,K,D}
418 264af2bb646f Switch to using * instead of calling the constructor Jonatan Werpers <jonatan@werpers.com> parents: 417 diff changeset	115 apply_transpose(c.t2, c.t1'*v, I...)
412 d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 402 diff changeset	116 end
190 8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	117
417 4c6604b7d990 Add dimension checking in the constructor Jonatan Werpers <jonatan@werpers.com> parents: 415 diff changeset	118 Base.@propagate_inbounds Base.:∘(s::TensorMapping, t::TensorMapping) = TensorMappingComposition(s,t)
190 8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	119
377 8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	120 """
393 b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 392 diff changeset	121 LazyLinearMap{T,R,D,...}(A, range_indicies, domain_indicies)
377 8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	122
8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	123 TensorMapping defined by the AbstractArray A. `range_indicies` and `domain_indicies` define which indicies of A should
400 c640f37d1c74 Update documentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 399 diff changeset	124 be considerd the range and domain of the TensorMapping. Each set of indices must be ordered in ascending order.
394 7ad644d112de Expand tests for LazyLinearMap and update its documentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 393 diff changeset	125
400 c640f37d1c74 Update documentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 399 diff changeset	126 For instance, if A is a m x n matrix, and range_size = (1,), domain_size = (2,), then the LazyLinearMap performs the
c640f37d1c74 Update documentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 399 diff changeset	127 standard matrix-vector product on vectors of size n.
377 8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	128 """
8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	129 struct LazyLinearMap{T,R,D, RD, AA<:AbstractArray{T,RD}} <: TensorMapping{T,R,D}
8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	130 A::AA
8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	131 range_indicies::NTuple{R,Int}
8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	132 domain_indicies::NTuple{D,Int}
399 3b4b1758a8ad Add a check for permuted indecies. Jonatan Werpers <jonatan@werpers.com> parents: 394 diff changeset	133
3b4b1758a8ad Add a check for permuted indecies. Jonatan Werpers <jonatan@werpers.com> parents: 394 diff changeset	134 function LazyLinearMap(A::AA, range_indicies::NTuple{R,Int}, domain_indicies::NTuple{D,Int}) where {T,R,D, RD, AA<:AbstractArray{T,RD}}
3b4b1758a8ad Add a check for permuted indecies. Jonatan Werpers <jonatan@werpers.com> parents: 394 diff changeset	135 if !issorted(range_indicies) \|\| !issorted(domain_indicies)
3b4b1758a8ad Add a check for permuted indecies. Jonatan Werpers <jonatan@werpers.com> parents: 394 diff changeset	136 throw(DomainError("range_indicies and domain_indicies must be sorted in ascending order"))
3b4b1758a8ad Add a check for permuted indecies. Jonatan Werpers <jonatan@werpers.com> parents: 394 diff changeset	137 end
3b4b1758a8ad Add a check for permuted indecies. Jonatan Werpers <jonatan@werpers.com> parents: 394 diff changeset	138
3b4b1758a8ad Add a check for permuted indecies. Jonatan Werpers <jonatan@werpers.com> parents: 394 diff changeset	139 return new{T,R,D,RD,AA}(A,range_indicies,domain_indicies)
3b4b1758a8ad Add a check for permuted indecies. Jonatan Werpers <jonatan@werpers.com> parents: 394 diff changeset	140 end
377 8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	141 end
8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	142
392 418cfd945715 Fix bug in range_size and domain_size for LazyLinearMap and expand the test Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 377 diff changeset	143 range_size(llm::LazyLinearMap) = size(llm.A)[[llm.range_indicies...]]
418cfd945715 Fix bug in range_size and domain_size for LazyLinearMap and expand the test Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 377 diff changeset	144 domain_size(llm::LazyLinearMap) = size(llm.A)[[llm.domain_indicies...]]
377 8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	145
943 fb060e98ac0a Remove more type assertions Jonatan Werpers <jonatan@werpers.com> parents: 942 diff changeset	146 function apply(llm::LazyLinearMap{T,R,D}, v::AbstractArray{<:Any,D}, I::Vararg{Any,R}) where {T,R,D}
377 8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	147 view_index = ntuple(i->:,ndims(llm.A))
8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	148 for i ∈ 1:R
8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	149 view_index = Base.setindex(view_index, Int(I[i]), llm.range_indicies[i])
8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	150 end
8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	151 A_view = @view llm.A[view_index...]
8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	152 return sum(A_view.*v)
8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	153 end
411 4aa59af074ef Add implementation of apply_transpose for LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 402 diff changeset	154
943 fb060e98ac0a Remove more type assertions Jonatan Werpers <jonatan@werpers.com> parents: 942 diff changeset	155 function apply_transpose(llm::LazyLinearMap{T,R,D}, v::AbstractArray{<:Any,R}, I::Vararg{Any,D}) where {T,R,D}
411 4aa59af074ef Add implementation of apply_transpose for LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 402 diff changeset	156 apply(LazyLinearMap(llm.A, llm.domain_indicies, llm.range_indicies), v, I...)
4aa59af074ef Add implementation of apply_transpose for LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 402 diff changeset	157 end
429 46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	158
46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	159
46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	160 """
434 648a36ebac99 Change name from LazyIdentity to IdentityMapping Jonatan Werpers <jonatan@werpers.com> parents: 433 diff changeset	161 IdentityMapping{T,D} <: TensorMapping{T,D,D}
429 46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	162
46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	163 The lazy identity TensorMapping for a given size. Usefull for building up higher dimensional tensor mappings from lower
46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	164 dimensional ones through outer products. Also used in the Implementation for InflatedTensorMapping.
46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	165 """
434 648a36ebac99 Change name from LazyIdentity to IdentityMapping Jonatan Werpers <jonatan@werpers.com> parents: 433 diff changeset	166 struct IdentityMapping{T,D} <: TensorMapping{T,D,D}
429 46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	167 size::NTuple{D,Int}
46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	168 end
46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	169
434 648a36ebac99 Change name from LazyIdentity to IdentityMapping Jonatan Werpers <jonatan@werpers.com> parents: 433 diff changeset	170 IdentityMapping{T}(size::NTuple{D,Int}) where {T,D} = IdentityMapping{T,D}(size)
441 552189707fbe Add convenience constructor Jonatan Werpers <jonatan@werpers.com> parents: 439 diff changeset	171 IdentityMapping{T}(size::Vararg{Int,D}) where {T,D} = IdentityMapping{T,D}(size)
442 e70e47fbfa7c Add another convenience constructor Jonatan Werpers <jonatan@werpers.com> parents: 441 diff changeset	172 IdentityMapping(size::Vararg{Int,D}) where D = IdentityMapping{Float64,D}(size)
429 46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	173
434 648a36ebac99 Change name from LazyIdentity to IdentityMapping Jonatan Werpers <jonatan@werpers.com> parents: 433 diff changeset	174 range_size(tmi::IdentityMapping) = tmi.size
648a36ebac99 Change name from LazyIdentity to IdentityMapping Jonatan Werpers <jonatan@werpers.com> parents: 433 diff changeset	175 domain_size(tmi::IdentityMapping) = tmi.size
429 46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	176
943 fb060e98ac0a Remove more type assertions Jonatan Werpers <jonatan@werpers.com> parents: 942 diff changeset	177 apply(tmi::IdentityMapping{T,D}, v::AbstractArray{<:Any,D}, I::Vararg{Any,D}) where {T,D} = v[I...]
fb060e98ac0a Remove more type assertions Jonatan Werpers <jonatan@werpers.com> parents: 942 diff changeset	178 apply_transpose(tmi::IdentityMapping{T,D}, v::AbstractArray{<:Any,D}, I::Vararg{Any,D}) where {T,D} = v[I...]
429 46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	179
479 95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 465 diff changeset	180 """
838 76e5682d0e52 Fix a bunch of docstring mistakes Jonatan Werpers <jonatan@werpers.com> parents: 625 diff changeset	181 Base.:∘(tm, tmi)
76e5682d0e52 Fix a bunch of docstring mistakes Jonatan Werpers <jonatan@werpers.com> parents: 625 diff changeset	182 Base.:∘(tmi, tm)
479 95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 465 diff changeset	183
480 c1a366331e75 Fix typo in docs Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 479 diff changeset	184 Composes a `Tensormapping` `tm` with an `IdentityMapping` `tmi`, by returning `tm`
479 95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 465 diff changeset	185 """
95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 465 diff changeset	186 @inline function Base.:∘(tm::TensorMapping{T,R,D}, tmi::IdentityMapping{T,D}) where {T,R,D}
485 4b49f03bdb98 Switch from DimensionMismatch to SizeMismatch for boundschecks on compositions Jonatan Werpers <jonatan@werpers.com> parents: 484 diff changeset	187 @boundscheck check_domain_size(tm, range_size(tmi))
479 95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 465 diff changeset	188 return tm
95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 465 diff changeset	189 end
95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 465 diff changeset	190
95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 465 diff changeset	191 @inline function Base.:∘(tmi::IdentityMapping{T,R}, tm::TensorMapping{T,R,D}) where {T,R,D}
485 4b49f03bdb98 Switch from DimensionMismatch to SizeMismatch for boundschecks on compositions Jonatan Werpers <jonatan@werpers.com> parents: 484 diff changeset	192 @boundscheck check_domain_size(tmi, range_size(tm))
479 95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 465 diff changeset	193 return tm
95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 465 diff changeset	194 end
95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 465 diff changeset	195 # Specialization for the case where tm is an IdentityMapping. Required to resolve ambiguity.
95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 465 diff changeset	196 @inline function Base.:∘(tm::IdentityMapping{T,D}, tmi::IdentityMapping{T,D}) where {T,D}
485 4b49f03bdb98 Switch from DimensionMismatch to SizeMismatch for boundschecks on compositions Jonatan Werpers <jonatan@werpers.com> parents: 484 diff changeset	197 @boundscheck check_domain_size(tm, range_size(tmi))
479 95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 465 diff changeset	198 return tmi
95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 465 diff changeset	199 end
95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 465 diff changeset	200
460 acffa3a1395d Move part of the documentation to the constructor Jonatan Werpers <jonatan@werpers.com> parents: 459 diff changeset	201
459 11e58c49fb46 Add some documentation Jonatan Werpers <jonatan@werpers.com> parents: 458 diff changeset	202 """
11e58c49fb46 Add some documentation Jonatan Werpers <jonatan@werpers.com> parents: 458 diff changeset	203 InflatedTensorMapping{T,R,D} <: TensorMapping{T,R,D}
11e58c49fb46 Add some documentation Jonatan Werpers <jonatan@werpers.com> parents: 458 diff changeset	204
11e58c49fb46 Add some documentation Jonatan Werpers <jonatan@werpers.com> parents: 458 diff changeset	205 An inflated `TensorMapping` with dimensions added before and afer its actual dimensions.
11e58c49fb46 Add some documentation Jonatan Werpers <jonatan@werpers.com> parents: 458 diff changeset	206 """
449 14d60de71b72 Change to concrete fields in the type Jonatan Werpers <jonatan@werpers.com> parents: 448 diff changeset	207 struct InflatedTensorMapping{T,R,D,D_before,R_middle,D_middle,D_after, TM<:TensorMapping{T,R_middle,D_middle}} <: TensorMapping{T,R,D}
448 912ae510dec9 Use new name of IdentityMapping Jonatan Werpers <jonatan@werpers.com> parents: 447 diff changeset	208 before::IdentityMapping{T,D_before}
449 14d60de71b72 Change to concrete fields in the type Jonatan Werpers <jonatan@werpers.com> parents: 448 diff changeset	209 tm::TM
448 912ae510dec9 Use new name of IdentityMapping Jonatan Werpers <jonatan@werpers.com> parents: 447 diff changeset	210 after::IdentityMapping{T,D_after}
446 904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	211
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	212 function InflatedTensorMapping(before, tm::TensorMapping{T}, after) where T
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	213 R_before = range_dim(before)
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	214 R_middle = range_dim(tm)
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	215 R_after = range_dim(after)
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	216 R = R_before+R_middle+R_after
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	217
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	218 D_before = domain_dim(before)
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	219 D_middle = domain_dim(tm)
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	220 D_after = domain_dim(after)
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	221 D = D_before+D_middle+D_after
449 14d60de71b72 Change to concrete fields in the type Jonatan Werpers <jonatan@werpers.com> parents: 448 diff changeset	222 return new{T,R,D,D_before,R_middle,D_middle,D_after, typeof(tm)}(before, tm, after)
448 912ae510dec9 Use new name of IdentityMapping Jonatan Werpers <jonatan@werpers.com> parents: 447 diff changeset	223 end
446 904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	224 end
461 a0e40d16ba0e Better documentation Jonatan Werpers <jonatan@werpers.com> parents: 460 diff changeset	225 """
464 73882867c121 Align documentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 463 diff changeset	226 InflatedTensorMapping(before, tm, after)
465 f270d82fc9ad Change documention in an attempt to follow the guidelines Jonatan Werpers <jonatan@werpers.com> parents: 464 diff changeset	227 InflatedTensorMapping(before,tm)
f270d82fc9ad Change documention in an attempt to follow the guidelines Jonatan Werpers <jonatan@werpers.com> parents: 464 diff changeset	228 InflatedTensorMapping(tm,after)
461 a0e40d16ba0e Better documentation Jonatan Werpers <jonatan@werpers.com> parents: 460 diff changeset	229
a0e40d16ba0e Better documentation Jonatan Werpers <jonatan@werpers.com> parents: 460 diff changeset	230 The outer product of `before`, `tm` and `after`, where `before` and `after` are `IdentityMapping`s.
465 f270d82fc9ad Change documention in an attempt to follow the guidelines Jonatan Werpers <jonatan@werpers.com> parents: 464 diff changeset	231
f270d82fc9ad Change documention in an attempt to follow the guidelines Jonatan Werpers <jonatan@werpers.com> parents: 464 diff changeset	232 If one of `before` or `after` is left out, a 0-dimensional `IdentityMapping` is used as the default value.
493 df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 487 diff changeset	233
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 487 diff changeset	234 If `tm` already is an `InflatedTensorMapping`, `before` and `after` will be extended instead of
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 487 diff changeset	235 creating a nested `InflatedTensorMapping`.
461 a0e40d16ba0e Better documentation Jonatan Werpers <jonatan@werpers.com> parents: 460 diff changeset	236 """
464 73882867c121 Align documentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 463 diff changeset	237 InflatedTensorMapping(::IdentityMapping, ::TensorMapping, ::IdentityMapping)
493 df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 487 diff changeset	238
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 487 diff changeset	239 function InflatedTensorMapping(before, itm::InflatedTensorMapping, after)
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 487 diff changeset	240 return InflatedTensorMapping(
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 487 diff changeset	241 IdentityMapping(before.size..., itm.before.size...),
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 487 diff changeset	242 itm.tm,
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 487 diff changeset	243 IdentityMapping(itm.after.size..., after.size...),
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 487 diff changeset	244 )
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 487 diff changeset	245 end
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 487 diff changeset	246
462 c364e2908c6e Implement InflatedTensorMapping where one of before and after are missing. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 458 diff changeset	247 InflatedTensorMapping(before::IdentityMapping, tm::TensorMapping{T}) where T = InflatedTensorMapping(before,tm,IdentityMapping{T}())
c364e2908c6e Implement InflatedTensorMapping where one of before and after are missing. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 458 diff changeset	248 InflatedTensorMapping(tm::TensorMapping{T}, after::IdentityMapping) where T = InflatedTensorMapping(IdentityMapping{T}(),tm,after)
465 f270d82fc9ad Change documention in an attempt to follow the guidelines Jonatan Werpers <jonatan@werpers.com> parents: 464 diff changeset	249 # Resolve ambiguity between the two previous methods
462 c364e2908c6e Implement InflatedTensorMapping where one of before and after are missing. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 458 diff changeset	250 InflatedTensorMapping(I1::IdentityMapping{T}, I2::IdentityMapping{T}) where T = InflatedTensorMapping(I1,I2,IdentityMapping{T}())
446 904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	251
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	252 # TODO: Implement some pretty printing in terms of ⊗. E.g InflatedTensorMapping(I(3),B,I(2)) -> I(3)⊗B⊗I(2)
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	253
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	254 function range_size(itm::InflatedTensorMapping)
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	255 return flatten_tuple(
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	256 range_size(itm.before),
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	257 range_size(itm.tm),
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	258 range_size(itm.after),
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	259 )
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	260 end
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	261
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	262 function domain_size(itm::InflatedTensorMapping)
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	263 return flatten_tuple(
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	264 domain_size(itm.before),
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	265 domain_size(itm.tm),
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	266 domain_size(itm.after),
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	267 )
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	268 end
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	269
943 fb060e98ac0a Remove more type assertions Jonatan Werpers <jonatan@werpers.com> parents: 942 diff changeset	270 function apply(itm::InflatedTensorMapping{T,R,D}, v::AbstractArray{<:Any,D}, I::Vararg{Any,R}) where {T,R,D}
538 848dec405332 Change variable names in apply and apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 537 diff changeset	271 dim_before = range_dim(itm.before)
848dec405332 Change variable names in apply and apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 537 diff changeset	272 dim_domain = domain_dim(itm.tm)
848dec405332 Change variable names in apply and apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 537 diff changeset	273 dim_range = range_dim(itm.tm)
848dec405332 Change variable names in apply and apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 537 diff changeset	274 dim_after = range_dim(itm.after)
520 fe86ac896377 Start refactoring split index and apply to accomodate future addition of apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 501 diff changeset	275
538 848dec405332 Change variable names in apply and apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 537 diff changeset	276 view_index, inner_index = split_index(Val(dim_before), Val(dim_domain), Val(dim_range), Val(dim_after), I...)
446 904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	277
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	278 v_inner = view(v, view_index...)
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	279 return apply(itm.tm, v_inner, inner_index...)
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	280 end
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	281
943 fb060e98ac0a Remove more type assertions Jonatan Werpers <jonatan@werpers.com> parents: 942 diff changeset	282 function apply_transpose(itm::InflatedTensorMapping{T,R,D}, v::AbstractArray{<:Any,R}, I::Vararg{Any,D}) where {T,R,D}
538 848dec405332 Change variable names in apply and apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 537 diff changeset	283 dim_before = range_dim(itm.before)
848dec405332 Change variable names in apply and apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 537 diff changeset	284 dim_domain = domain_dim(itm.tm)
848dec405332 Change variable names in apply and apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 537 diff changeset	285 dim_range = range_dim(itm.tm)
848dec405332 Change variable names in apply and apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 537 diff changeset	286 dim_after = range_dim(itm.after)
526 be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 525 diff changeset	287
538 848dec405332 Change variable names in apply and apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 537 diff changeset	288 view_index, inner_index = split_index(Val(dim_before), Val(dim_range), Val(dim_domain), Val(dim_after), I...)
526 be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 525 diff changeset	289
be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 525 diff changeset	290 v_inner = view(v, view_index...)
be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 525 diff changeset	291 return apply_transpose(itm.tm, v_inner, inner_index...)
be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 525 diff changeset	292 end
be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 525 diff changeset	293
446 904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	294
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	295 """
533 aac7cc1fa79a Try to improve the naming in split_index() Jonatan Werpers <jonatan@werpers.com> parents: 532 diff changeset	296 split_index(::Val{dim_before}, ::Val{dim_view}, ::Val{dim_index}, ::Val{dim_after}, I...)
446 904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	297
533 aac7cc1fa79a Try to improve the naming in split_index() Jonatan Werpers <jonatan@werpers.com> parents: 532 diff changeset	298 Splits the multi-index `I` into two parts. One part which is expected to be
aac7cc1fa79a Try to improve the naming in split_index() Jonatan Werpers <jonatan@werpers.com> parents: 532 diff changeset	299 used as a view, and one which is expected to be used as an index.
446 904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	300 Eg.
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	301 ```
537 41e82a5d4d48 Docs touch up Jonatan Werpers <jonatan@werpers.com> parents: 533 diff changeset	302 split_index(Val(1),Val(3),Val(2),Val(1),(1,2,3,4)) -> (1,:,:,:,4), (2,3)
446 904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	303 ```
520 fe86ac896377 Start refactoring split index and apply to accomodate future addition of apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 501 diff changeset	304
533 aac7cc1fa79a Try to improve the naming in split_index() Jonatan Werpers <jonatan@werpers.com> parents: 532 diff changeset	305 `dim_view` controls how many colons are in the view, and `dim_index` controls
aac7cc1fa79a Try to improve the naming in split_index() Jonatan Werpers <jonatan@werpers.com> parents: 532 diff changeset	306 how many elements are extracted from the middle.
aac7cc1fa79a Try to improve the naming in split_index() Jonatan Werpers <jonatan@werpers.com> parents: 532 diff changeset	307 `dim_before` and `dim_after` decides the length of the index parts before and after the colons in the view index.
531 36dfc57e8e0b Write out some more documentation for split_index Jonatan Werpers <jonatan@werpers.com> parents: 530 diff changeset	308
533 aac7cc1fa79a Try to improve the naming in split_index() Jonatan Werpers <jonatan@werpers.com> parents: 532 diff changeset	309 Arguments should satisfy `length(I) == dim_before+B_domain+dim_after`.
520 fe86ac896377 Start refactoring split index and apply to accomodate future addition of apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 501 diff changeset	310
531 36dfc57e8e0b Write out some more documentation for split_index Jonatan Werpers <jonatan@werpers.com> parents: 530 diff changeset	311 The returned values satisfy
533 aac7cc1fa79a Try to improve the naming in split_index() Jonatan Werpers <jonatan@werpers.com> parents: 532 diff changeset	312 * `length(view_index) == dim_before + dim_view + dim_after`
aac7cc1fa79a Try to improve the naming in split_index() Jonatan Werpers <jonatan@werpers.com> parents: 532 diff changeset	313 * `length(I_middle) == dim_index`
446 904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	314 """
533 aac7cc1fa79a Try to improve the naming in split_index() Jonatan Werpers <jonatan@werpers.com> parents: 532 diff changeset	315 function split_index(::Val{dim_before}, ::Val{dim_view}, ::Val{dim_index}, ::Val{dim_after}, I...) where {dim_before,dim_view, dim_index,dim_after}
aac7cc1fa79a Try to improve the naming in split_index() Jonatan Werpers <jonatan@werpers.com> parents: 532 diff changeset	316 I_before, I_middle, I_after = split_tuple(I, Val(dim_before), Val(dim_index))
446 904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	317
533 aac7cc1fa79a Try to improve the naming in split_index() Jonatan Werpers <jonatan@werpers.com> parents: 532 diff changeset	318 view_index = (I_before..., ntuple((i)->:, dim_view)..., I_after...)
446 904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	319
520 fe86ac896377 Start refactoring split index and apply to accomodate future addition of apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 501 diff changeset	320 return view_index, I_middle
446 904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	321 end
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	322
456 8f4c31e06689 Add docs and tests for slice_tuple Jonatan Werpers <jonatan@werpers.com> parents: 455 diff changeset	323 # TODO: Can this be replaced by something more elegant while still being type stable? 2020-10-21
8f4c31e06689 Add docs and tests for slice_tuple Jonatan Werpers <jonatan@werpers.com> parents: 455 diff changeset	324 # See:
8f4c31e06689 Add docs and tests for slice_tuple Jonatan Werpers <jonatan@werpers.com> parents: 455 diff changeset	325 # https://github.com/JuliaLang/julia/issues/34884
8f4c31e06689 Add docs and tests for slice_tuple Jonatan Werpers <jonatan@werpers.com> parents: 455 diff changeset	326 # https://github.com/JuliaLang/julia/issues/30386
8f4c31e06689 Add docs and tests for slice_tuple Jonatan Werpers <jonatan@werpers.com> parents: 455 diff changeset	327 """
8f4c31e06689 Add docs and tests for slice_tuple Jonatan Werpers <jonatan@werpers.com> parents: 455 diff changeset	328 slice_tuple(t, Val(l), Val(u))
8f4c31e06689 Add docs and tests for slice_tuple Jonatan Werpers <jonatan@werpers.com> parents: 455 diff changeset	329
8f4c31e06689 Add docs and tests for slice_tuple Jonatan Werpers <jonatan@werpers.com> parents: 455 diff changeset	330 Get a slice of a tuple in a type stable way.
838 76e5682d0e52 Fix a bunch of docstring mistakes Jonatan Werpers <jonatan@werpers.com> parents: 625 diff changeset	331 Equivalent to `t[l:u]` but type stable.
456 8f4c31e06689 Add docs and tests for slice_tuple Jonatan Werpers <jonatan@werpers.com> parents: 455 diff changeset	332 """
455 b86312d14873 Make split_index type stable Jonatan Werpers <jonatan@werpers.com> parents: 452 diff changeset	333 function slice_tuple(t,::Val{L},::Val{U}) where {L,U}
b86312d14873 Make split_index type stable Jonatan Werpers <jonatan@werpers.com> parents: 452 diff changeset	334 return ntuple(i->t[i+L-1], U-L+1)
b86312d14873 Make split_index type stable Jonatan Werpers <jonatan@werpers.com> parents: 452 diff changeset	335 end
b86312d14873 Make split_index type stable Jonatan Werpers <jonatan@werpers.com> parents: 452 diff changeset	336
457 8fb6a5611c7a Add tests and docs for flatten_tuple Jonatan Werpers <jonatan@werpers.com> parents: 456 diff changeset	337 """
532 588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	338 split_tuple(t::Tuple{...}, ::Val{M}) where {N,M}
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	339
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	340 Split the tuple `t` into two parts. the first part is `M` long.
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	341 E.g
838 76e5682d0e52 Fix a bunch of docstring mistakes Jonatan Werpers <jonatan@werpers.com> parents: 625 diff changeset	342 ```julia
532 588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	343 split_tuple((1,2,3,4),Val(3)) -> (1,2,3), (4,)
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	344 ```
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	345 """
625 1c512e796c6d Allow split_tuple to split tuples containing different types. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 562 diff changeset	346 function split_tuple(t::NTuple{N,Any},::Val{M}) where {N,M}
532 588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	347 return slice_tuple(t,Val(1), Val(M)), slice_tuple(t,Val(M+1), Val(N))
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	348 end
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	349
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	350 """
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	351 split_tuple(t::Tuple{...},::Val{M},::Val{K}) where {N,M,K}
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	352
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	353 Same as `split_tuple(t::NTuple{N},::Val{M})` but splits the tuple in three parts. With the first
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	354 two parts having lenght `M` and `K`.
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	355 """
625 1c512e796c6d Allow split_tuple to split tuples containing different types. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 562 diff changeset	356 function split_tuple(t::NTuple{N,Any},::Val{M},::Val{K}) where {N,M,K}
532 588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	357 p1, tail = split_tuple(t, Val(M))
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	358 p2, p3 = split_tuple(tail, Val(K))
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	359 return p1,p2,p3
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	360 end
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	361
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	362
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	363 """
457 8fb6a5611c7a Add tests and docs for flatten_tuple Jonatan Werpers <jonatan@werpers.com> parents: 456 diff changeset	364 flatten_tuple(t)
8fb6a5611c7a Add tests and docs for flatten_tuple Jonatan Werpers <jonatan@werpers.com> parents: 456 diff changeset	365
8fb6a5611c7a Add tests and docs for flatten_tuple Jonatan Werpers <jonatan@werpers.com> parents: 456 diff changeset	366 Takes a nested tuple and flattens the whole structure
8fb6a5611c7a Add tests and docs for flatten_tuple Jonatan Werpers <jonatan@werpers.com> parents: 456 diff changeset	367 """
446 904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	368 flatten_tuple(t::NTuple{N, Number} where N) = t
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	369 flatten_tuple(t::Tuple) = ((flatten_tuple.(t)...)...,) # simplify?
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	370 flatten_tuple(ts::Vararg) = flatten_tuple(ts)
487 6a6b7eaf9edf Move exception handling to end of file and update error message. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 485 diff changeset	371
838 76e5682d0e52 Fix a bunch of docstring mistakes Jonatan Werpers <jonatan@werpers.com> parents: 625 diff changeset	372 @doc raw"""
76e5682d0e52 Fix a bunch of docstring mistakes Jonatan Werpers <jonatan@werpers.com> parents: 625 diff changeset	373 LazyOuterProduct(tms...)
468 a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	374
501 4b9d124fe984 Fix typo in documentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 499 diff changeset	375 Creates a `TensorMappingComposition` for the outerproduct of `tms...`.
468 a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	376 This is done by separating the outer product into regular products of outer products involving only identity mappings and one non-identity mapping.
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	377
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	378 First let
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	379 ```math
838 76e5682d0e52 Fix a bunch of docstring mistakes Jonatan Werpers <jonatan@werpers.com> parents: 625 diff changeset	380 \begin{aligned}
76e5682d0e52 Fix a bunch of docstring mistakes Jonatan Werpers <jonatan@werpers.com> parents: 625 diff changeset	381 A &= A_{I,J} \\
76e5682d0e52 Fix a bunch of docstring mistakes Jonatan Werpers <jonatan@werpers.com> parents: 625 diff changeset	382 B &= B_{M,N} \\
76e5682d0e52 Fix a bunch of docstring mistakes Jonatan Werpers <jonatan@werpers.com> parents: 625 diff changeset	383 C &= C_{P,Q} \\
76e5682d0e52 Fix a bunch of docstring mistakes Jonatan Werpers <jonatan@werpers.com> parents: 625 diff changeset	384 \end{aligned}
468 a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	385 ```
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	386
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	387 where ``I``, ``M``, ``P`` are multi-indexes for the ranges of ``A``, ``B``, ``C``, and ``J``, ``N``, ``Q`` are multi-indexes of the domains.
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	388
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	389 We use ``⊗`` to denote the outer product
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	390 ```math
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	391 (A⊗B)_{IM,JN} = A_{I,J}B_{M,N}
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	392 ```
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	393
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	394 We note that
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	395 ```math
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	396 A⊗B⊗C = (A⊗B⊗C)_{IMP,JNQ} = A_{I,J}B_{M,N}C_{P,Q}
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	397 ```
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	398 And that
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	399 ```math
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	400 A⊗B⊗C = (A⊗I_{\|M\|}⊗I_{\|P\|})(I_{\|J\|}⊗B⊗I_{\|P\|})(I_{\|J\|}⊗I_{\|N\|}⊗C)
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	401 ```
838 76e5682d0e52 Fix a bunch of docstring mistakes Jonatan Werpers <jonatan@werpers.com> parents: 625 diff changeset	402 where ``\|⋅\|`` of a multi-index is a vector of sizes for each dimension. ``I_v`` denotes the identity tensor of size ``v[i]`` in each direction
468 a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	403 To apply ``A⊗B⊗C`` we evaluate
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	404
838 76e5682d0e52 Fix a bunch of docstring mistakes Jonatan Werpers <jonatan@werpers.com> parents: 625 diff changeset	405 ```math
468 a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	406 (A⊗B⊗C)v = [(A⊗I_{\|M\|}⊗I_{\|P\|}) [(I_{\|J\|}⊗B⊗I_{\|P\|}) [(I_{\|J\|}⊗I_{\|N\|}⊗C)v]]]
838 76e5682d0e52 Fix a bunch of docstring mistakes Jonatan Werpers <jonatan@werpers.com> parents: 625 diff changeset	407 ```
468 a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	408 """
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	409 function LazyOuterProduct end
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	410
470 0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	411 function LazyOuterProduct(tm1::TensorMapping{T}, tm2::TensorMapping{T}) where T
0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	412 itm1 = InflatedTensorMapping(tm1, IdentityMapping{T}(range_size(tm2)))
0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	413 itm2 = InflatedTensorMapping(IdentityMapping{T}(domain_size(tm1)),tm2)
468 a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	414
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	415 return itm1∘itm2
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	416 end
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	417
491 2dc2eac27f75 Add special methods for Identity mappings Jonatan Werpers <jonatan@werpers.com> parents: 475 diff changeset	418 LazyOuterProduct(t1::IdentityMapping{T}, t2::IdentityMapping{T}) where T = IdentityMapping{T}(t1.size...,t2.size...)
2dc2eac27f75 Add special methods for Identity mappings Jonatan Werpers <jonatan@werpers.com> parents: 475 diff changeset	419 LazyOuterProduct(t1::TensorMapping, t2::IdentityMapping) = InflatedTensorMapping(t1, t2)
2dc2eac27f75 Add special methods for Identity mappings Jonatan Werpers <jonatan@werpers.com> parents: 475 diff changeset	420 LazyOuterProduct(t1::IdentityMapping, t2::TensorMapping) = InflatedTensorMapping(t1, t2)
2dc2eac27f75 Add special methods for Identity mappings Jonatan Werpers <jonatan@werpers.com> parents: 475 diff changeset	421
470 0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	422 LazyOuterProduct(tms::Vararg{TensorMapping}) = foldl(LazyOuterProduct, tms)
468 a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	423
474 3ad327378b2d Simplify implementation of \otimes Jonatan Werpers <jonatan@werpers.com> parents: 470 diff changeset	424 ⊗(a::TensorMapping, b::TensorMapping) = LazyOuterProduct(a,b)
468 a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	425
498 5a600ec40ccc Merge in default Jonatan Werpers <jonatan@werpers.com> parents: 491 493 diff changeset	426
487 6a6b7eaf9edf Move exception handling to end of file and update error message. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 485 diff changeset	427 function check_domain_size(tm::TensorMapping, sz)
6a6b7eaf9edf Move exception handling to end of file and update error message. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 485 diff changeset	428 if domain_size(tm) != sz
6a6b7eaf9edf Move exception handling to end of file and update error message. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 485 diff changeset	429 throw(SizeMismatch(tm,sz))
6a6b7eaf9edf Move exception handling to end of file and update error message. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 485 diff changeset	430 end
6a6b7eaf9edf Move exception handling to end of file and update error message. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 485 diff changeset	431 end
6a6b7eaf9edf Move exception handling to end of file and update error message. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 485 diff changeset	432
6a6b7eaf9edf Move exception handling to end of file and update error message. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 485 diff changeset	433 struct SizeMismatch <: Exception
6a6b7eaf9edf Move exception handling to end of file and update error message. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 485 diff changeset	434 tm::TensorMapping
6a6b7eaf9edf Move exception handling to end of file and update error message. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 485 diff changeset	435 sz
6a6b7eaf9edf Move exception handling to end of file and update error message. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 485 diff changeset	436 end
6a6b7eaf9edf Move exception handling to end of file and update error message. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 485 diff changeset	437
6a6b7eaf9edf Move exception handling to end of file and update error message. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 485 diff changeset	438 function Base.showerror(io::IO, err::SizeMismatch)
6a6b7eaf9edf Move exception handling to end of file and update error message. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 485 diff changeset	439 print(io, "SizeMismatch: ")
6a6b7eaf9edf Move exception handling to end of file and update error message. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 485 diff changeset	440 print(io, "domain size $(domain_size(err.tm)) of TensorMapping not matching size $(err.sz)")
6a6b7eaf9edf Move exception handling to end of file and update error message. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 485 diff changeset	441 end
919 b41180efb6c2 Start refactor to improve type stability for apply(op,::TensorApplication,...) Jonatan Werpers <jonatan@werpers.com> parents: 838 diff changeset	442
b41180efb6c2 Start refactor to improve type stability for apply(op,::TensorApplication,...) Jonatan Werpers <jonatan@werpers.com> parents: 838 diff changeset	443 function apply_with_region(op, v, boundary_width::Integer, dim_size::Integer, i)
b41180efb6c2 Start refactor to improve type stability for apply(op,::TensorApplication,...) Jonatan Werpers <jonatan@werpers.com> parents: 838 diff changeset	444 if 0 < i <= boundary_width
b41180efb6c2 Start refactor to improve type stability for apply(op,::TensorApplication,...) Jonatan Werpers <jonatan@werpers.com> parents: 838 diff changeset	445 return LazyTensors.apply(op,v,Index(i,Lower))
b41180efb6c2 Start refactor to improve type stability for apply(op,::TensorApplication,...) Jonatan Werpers <jonatan@werpers.com> parents: 838 diff changeset	446 elseif boundary_width < i <= dim_size-boundary_width
b41180efb6c2 Start refactor to improve type stability for apply(op,::TensorApplication,...) Jonatan Werpers <jonatan@werpers.com> parents: 838 diff changeset	447 return LazyTensors.apply(op,v,Index(i,Interior))
b41180efb6c2 Start refactor to improve type stability for apply(op,::TensorApplication,...) Jonatan Werpers <jonatan@werpers.com> parents: 838 diff changeset	448 elseif dim_size-boundary_width < i <= dim_size
b41180efb6c2 Start refactor to improve type stability for apply(op,::TensorApplication,...) Jonatan Werpers <jonatan@werpers.com> parents: 838 diff changeset	449 return LazyTensors.apply(op,v,Index(i,Upper))
b41180efb6c2 Start refactor to improve type stability for apply(op,::TensorApplication,...) Jonatan Werpers <jonatan@werpers.com> parents: 838 diff changeset	450 else
b41180efb6c2 Start refactor to improve type stability for apply(op,::TensorApplication,...) Jonatan Werpers <jonatan@werpers.com> parents: 838 diff changeset	451 error("Bounds error") # TODO: Make this more standard
b41180efb6c2 Start refactor to improve type stability for apply(op,::TensorApplication,...) Jonatan Werpers <jonatan@werpers.com> parents: 838 diff changeset	452 end
b41180efb6c2 Start refactor to improve type stability for apply(op,::TensorApplication,...) Jonatan Werpers <jonatan@werpers.com> parents: 838 diff changeset	453 end
b41180efb6c2 Start refactor to improve type stability for apply(op,::TensorApplication,...) Jonatan Werpers <jonatan@werpers.com> parents: 838 diff changeset	454 # TBD: Can these methods be merge by having a function as an arguement instead?
b41180efb6c2 Start refactor to improve type stability for apply(op,::TensorApplication,...) Jonatan Werpers <jonatan@werpers.com> parents: 838 diff changeset	455 # TODO: Add inference test that show where things break and how this rewrite fixes it.
b41180efb6c2 Start refactor to improve type stability for apply(op,::TensorApplication,...) Jonatan Werpers <jonatan@werpers.com> parents: 838 diff changeset	456 function apply_transpose_with_region(op, v, boundary_width::Integer, dim_size::Integer, i)
b41180efb6c2 Start refactor to improve type stability for apply(op,::TensorApplication,...) Jonatan Werpers <jonatan@werpers.com> parents: 838 diff changeset	457 if 0 < i <= boundary_width
b41180efb6c2 Start refactor to improve type stability for apply(op,::TensorApplication,...) Jonatan Werpers <jonatan@werpers.com> parents: 838 diff changeset	458 return LazyTensors.apply_transpose(op,v,Index(i,Lower))
b41180efb6c2 Start refactor to improve type stability for apply(op,::TensorApplication,...) Jonatan Werpers <jonatan@werpers.com> parents: 838 diff changeset	459 elseif boundary_width < i <= dim_size-boundary_width
b41180efb6c2 Start refactor to improve type stability for apply(op,::TensorApplication,...) Jonatan Werpers <jonatan@werpers.com> parents: 838 diff changeset	460 return LazyTensors.apply_transpose(op,v,Index(i,Interior))
b41180efb6c2 Start refactor to improve type stability for apply(op,::TensorApplication,...) Jonatan Werpers <jonatan@werpers.com> parents: 838 diff changeset	461 elseif dim_size-boundary_width < i <= dim_size
b41180efb6c2 Start refactor to improve type stability for apply(op,::TensorApplication,...) Jonatan Werpers <jonatan@werpers.com> parents: 838 diff changeset	462 return LazyTensors.apply_transpose(op,v,Index(i,Upper))
b41180efb6c2 Start refactor to improve type stability for apply(op,::TensorApplication,...) Jonatan Werpers <jonatan@werpers.com> parents: 838 diff changeset	463 else
b41180efb6c2 Start refactor to improve type stability for apply(op,::TensorApplication,...) Jonatan Werpers <jonatan@werpers.com> parents: 838 diff changeset	464 error("Bounds error") # TODO: Make this more standard
b41180efb6c2 Start refactor to improve type stability for apply(op,::TensorApplication,...) Jonatan Werpers <jonatan@werpers.com> parents: 838 diff changeset	465 end
b41180efb6c2 Start refactor to improve type stability for apply(op,::TensorApplication,...) Jonatan Werpers <jonatan@werpers.com> parents: 838 diff changeset	466 end

Mercurial > repos > public > sbplib_julia

annotate src/LazyTensors/lazy_tensor_operations.jl @ 971:bc12be1b1ae5 feature/variable_derivatives