sbplib_julia: src/LazyTensors/lazy_tensor

annotate src/LazyTensors/lazy_tensor_operations.jl @ 541:62d96e2cd165 refactor/tensor_index_coupling

Make the coupling between all the LazyTensors code and the Index type much weaker to make the module more flexible

author	Jonatan Werpers <jonatan@werpers.com>
date	Thu, 26 Nov 2020 21:35:34 +0100
parents	848dec405332
children	53828d3ed132

rev	line source
196 b3c252280a19 Move LazyArray and make LazyTensorMappingApplication and LazyElementwiseOperation subtypes of it Jonatan Werpers <jonatan@werpers.com> parents: 195 diff changeset	1 """
b3c252280a19 Move LazyArray and make LazyTensorMappingApplication and LazyElementwiseOperation subtypes of it Jonatan Werpers <jonatan@werpers.com> parents: 195 diff changeset	2 LazyTensorMappingApplication{T,R,D} <: LazyArray{T,R}
190 8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	3
8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	4 Struct for lazy application of a TensorMapping. Created using `*`.
8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	5
8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	6 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	7 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	8 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	9 """
378 946516954c85 Use concrete types in LazyTensorApplication Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	10 struct LazyTensorMappingApplication{T,R,D, TM<:TensorMapping{T,R,D}, AA<:AbstractArray{T,D}} <: LazyArray{T,R}
946516954c85 Use concrete types in LazyTensorApplication Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	11 t::TM
946516954c85 Use concrete types in LazyTensorApplication Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	12 o::AA
190 8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	13 end
348 7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 333 diff changeset	14 # TODO: Do boundschecking on creation!
190 8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	15 export LazyTensorMappingApplication
8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	16
379 de4746d6d126 Add some notes and todos Jonatan Werpers <jonatan@werpers.com> parents: 378 diff changeset	17 # TODO: Go through and remove unneccerary type parameters on functions
de4746d6d126 Add some notes and todos Jonatan Werpers <jonatan@werpers.com> parents: 378 diff changeset	18
541 62d96e2cd165 Make the coupling between all the LazyTensors code and the Index type much weaker to make the module more flexible Jonatan Werpers <jonatan@werpers.com> parents: 538 diff changeset	19 Base.getindex(ta::LazyTensorMappingApplication{T,R,D}, I::Vararg{Any,R}) where {T,R,D} = apply(ta.t, ta.o, I...)
359 ba46a952a450 Remove unnecessary type parameters Jonatan Werpers <jonatan@werpers.com> parents: 348 diff changeset	20 Base.size(ta::LazyTensorMappingApplication) = range_size(ta.t)
190 8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	21 # 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	22
421 2808c41f5efa Change defs of * for TensorApplication to fix bug Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	23 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	24 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	25 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	26
190 8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	27 # # 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	28 # # Should we overload some other infix binary opesrator?
190 8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	29 # →(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	30 # 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	31 # 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	32
194 d30d42a11566 Move all Lazy operation into the same module Jonatan Werpers <jonatan@werpers.com> parents: 190 diff changeset	33 """
197 a340fa91b1fc Move transpose def to the bottom of the file Jonatan Werpers <jonatan@werpers.com> parents: 196 diff changeset	34 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	35
a340fa91b1fc Move transpose def to the bottom of the file Jonatan Werpers <jonatan@werpers.com> parents: 196 diff changeset	36 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	37
a340fa91b1fc Move transpose def to the bottom of the file Jonatan Werpers <jonatan@werpers.com> parents: 196 diff changeset	38 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	39 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	40 the appropriate methods of `m`.
a340fa91b1fc Move transpose def to the bottom of the file Jonatan Werpers <jonatan@werpers.com> parents: 196 diff changeset	41 """
438 907b0510699f Change LazyTensorTranspose to have concrete fields Jonatan Werpers <jonatan@werpers.com> parents: 436 diff changeset	42 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	43 tm::TM
197 a340fa91b1fc Move transpose def to the bottom of the file Jonatan Werpers <jonatan@werpers.com> parents: 196 diff changeset	44 end
a340fa91b1fc Move transpose def to the bottom of the file Jonatan Werpers <jonatan@werpers.com> parents: 196 diff changeset	45 export LazyTensorMappingTranspose
a340fa91b1fc Move transpose def to the bottom of the file Jonatan Werpers <jonatan@werpers.com> parents: 196 diff changeset	46
a340fa91b1fc Move transpose def to the bottom of the file Jonatan Werpers <jonatan@werpers.com> parents: 196 diff changeset	47 # # 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	48 # 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	49 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	50 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	51
541 62d96e2cd165 Make the coupling between all the LazyTensors code and the Index type much weaker to make the module more flexible Jonatan Werpers <jonatan@werpers.com> parents: 538 diff changeset	52 apply(tmt::LazyTensorMappingTranspose{T,R,D}, v::AbstractArray{T,R}, I::Vararg{Any,D}) where {T,R,D} = apply_transpose(tmt.tm, v, I...)
62d96e2cd165 Make the coupling between all the LazyTensors code and the Index type much weaker to make the module more flexible Jonatan Werpers <jonatan@werpers.com> parents: 538 diff changeset	53 apply_transpose(tmt::LazyTensorMappingTranspose{T,R,D}, v::AbstractArray{T,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	54
359 ba46a952a450 Remove unnecessary type parameters Jonatan Werpers <jonatan@werpers.com> parents: 348 diff changeset	55 range_size(tmt::LazyTensorMappingTranspose) = domain_size(tmt.tm)
ba46a952a450 Remove unnecessary type parameters Jonatan Werpers <jonatan@werpers.com> parents: 348 diff changeset	56 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	57
210 2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 209 diff changeset	58
2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 209 diff changeset	59 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	60 tm1::T1
11010bb74260 Dispatch getindex for TensorMappingApplication on region indices. Dispatch apply Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 267 diff changeset	61 tm2::T2
210 2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 209 diff changeset	62
274 11010bb74260 Dispatch getindex for TensorMappingApplication on region indices. Dispatch apply Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 267 diff changeset	63 @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	64 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	65 end
2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 209 diff changeset	66 end
348 7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 333 diff changeset	67 # TODO: Boundschecking in constructor.
210 2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 209 diff changeset	68
541 62d96e2cd165 Make the coupling between all the LazyTensors code and the Index type much weaker to make the module more flexible Jonatan Werpers <jonatan@werpers.com> parents: 538 diff changeset	69 apply(tmBinOp::LazyTensorMappingBinaryOperation{:+,T,R,D}, v::AbstractArray{T,D}, I::Vararg{Any,R}) where {T,R,D} = apply(tmBinOp.tm1, v, I...) + apply(tmBinOp.tm2, v, I...)
62d96e2cd165 Make the coupling between all the LazyTensors code and the Index type much weaker to make the module more flexible Jonatan Werpers <jonatan@werpers.com> parents: 538 diff changeset	70 apply(tmBinOp::LazyTensorMappingBinaryOperation{:-,T,R,D}, v::AbstractArray{T,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	71
348 7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 333 diff changeset	72 range_size(tmBinOp::LazyTensorMappingBinaryOperation{Op,T,R,D}) where {Op,T,R,D} = range_size(tmBinOp.tm1)
7fe43d902a27 Start trying to change LazyTensors Jonatan Werpers <jonatan@werpers.com> parents: 333 diff changeset	73 domain_size(tmBinOp::LazyTensorMappingBinaryOperation{Op,T,R,D}) where {Op,T,R,D} = domain_size(tmBinOp.tm1)
210 2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 209 diff changeset	74
274 11010bb74260 Dispatch getindex for TensorMappingApplication on region indices. Dispatch apply Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 267 diff changeset	75 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	76 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	77
412 d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 402 diff changeset	78 """
d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 402 diff changeset	79 TensorMappingComposition{T,R,K,D}
210 2aa33d0eef90 Add ability to add and subtract TensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 209 diff changeset	80
412 d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 402 diff changeset	81 Lazily compose two TensorMappings, so that they can be handled as a single TensorMapping.
d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 402 diff changeset	82 """
d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 402 diff changeset	83 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	84 t1::TM1
d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 402 diff changeset	85 t2::TM2
190 8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	86
417 4c6604b7d990 Add dimension checking in the constructor Jonatan Werpers <jonatan@werpers.com> parents: 415 diff changeset	87 @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	88 @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	89 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	90 end
412 d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 402 diff changeset	91 end
d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 402 diff changeset	92 export TensorMappingComposition
190 8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	93
412 d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 402 diff changeset	94 range_size(tm::TensorMappingComposition) = range_size(tm.t1)
d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 402 diff changeset	95 domain_size(tm::TensorMappingComposition) = domain_size(tm.t2)
190 8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	96
412 d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 402 diff changeset	97 function apply(c::TensorMappingComposition{T,R,K,D}, v::AbstractArray{T,D}, I::Vararg{S,R} where S) where {T,R,K,D}
418 264af2bb646f Switch to using * instead of calling the constructor Jonatan Werpers <jonatan@werpers.com> parents: 417 diff changeset	98 apply(c.t1, c.t2*v, I...)
412 d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 402 diff changeset	99 end
190 8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	100
412 d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 402 diff changeset	101 function apply_transpose(c::TensorMappingComposition{T,R,K,D}, v::AbstractArray{T,R}, I::Vararg{S,D} where S) where {T,R,K,D}
418 264af2bb646f Switch to using * instead of calling the constructor Jonatan Werpers <jonatan@werpers.com> parents: 417 diff changeset	102 apply_transpose(c.t2, c.t1'*v, I...)
412 d94891b8dfca Start implementing TensorComposition Jonatan Werpers <jonatan@werpers.com> parents: 402 diff changeset	103 end
190 8964b3165097 Break LazyTensors.jl into several files Jonatan Werpers <jonatan@werpers.com> parents: diff changeset	104
417 4c6604b7d990 Add dimension checking in the constructor Jonatan Werpers <jonatan@werpers.com> parents: 415 diff changeset	105 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	106
377 8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	107 """
393 b14eacf823b6 Test applying LazyLinearMap to random vectors and matrices. Fix indentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 392 diff changeset	108 LazyLinearMap{T,R,D,...}(A, range_indicies, domain_indicies)
377 8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	109
8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	110 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	111 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	112
400 c640f37d1c74 Update documentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 399 diff changeset	113 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	114 standard matrix-vector product on vectors of size n.
377 8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	115 """
8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	116 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	117 A::AA
8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	118 range_indicies::NTuple{R,Int}
8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	119 domain_indicies::NTuple{D,Int}
399 3b4b1758a8ad Add a check for permuted indecies. Jonatan Werpers <jonatan@werpers.com> parents: 394 diff changeset	120
3b4b1758a8ad Add a check for permuted indecies. Jonatan Werpers <jonatan@werpers.com> parents: 394 diff changeset	121 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	122 if !issorted(range_indicies) \|\| !issorted(domain_indicies)
3b4b1758a8ad Add a check for permuted indecies. Jonatan Werpers <jonatan@werpers.com> parents: 394 diff changeset	123 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	124 end
3b4b1758a8ad Add a check for permuted indecies. Jonatan Werpers <jonatan@werpers.com> parents: 394 diff changeset	125
3b4b1758a8ad Add a check for permuted indecies. Jonatan Werpers <jonatan@werpers.com> parents: 394 diff changeset	126 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	127 end
377 8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	128 end
8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	129 export LazyLinearMap
8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	130
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	131 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	132 domain_size(llm::LazyLinearMap) = size(llm.A)[[llm.domain_indicies...]]
377 8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	133
541 62d96e2cd165 Make the coupling between all the LazyTensors code and the Index type much weaker to make the module more flexible Jonatan Werpers <jonatan@werpers.com> parents: 538 diff changeset	134 function apply(llm::LazyLinearMap{T,R,D}, v::AbstractArray{T,D}, I::Vararg{Any,R}) where {T,R,D}
377 8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	135 view_index = ntuple(i->:,ndims(llm.A))
8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	136 for i ∈ 1:R
8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	137 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	138 end
8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	139 A_view = @view llm.A[view_index...]
8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	140 return sum(A_view.*v)
8414c2334393 Start implementing LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 359 diff changeset	141 end
411 4aa59af074ef Add implementation of apply_transpose for LazyLinearMap Jonatan Werpers <jonatan@werpers.com> parents: 402 diff changeset	142
541 62d96e2cd165 Make the coupling between all the LazyTensors code and the Index type much weaker to make the module more flexible Jonatan Werpers <jonatan@werpers.com> parents: 538 diff changeset	143 function apply_transpose(llm::LazyLinearMap{T,R,D}, v::AbstractArray{T,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	144 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	145 end
429 46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	146
46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	147
46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	148 """
434 648a36ebac99 Change name from LazyIdentity to IdentityMapping Jonatan Werpers <jonatan@werpers.com> parents: 433 diff changeset	149 IdentityMapping{T,D} <: TensorMapping{T,D,D}
429 46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	150
46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	151 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	152 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	153 """
434 648a36ebac99 Change name from LazyIdentity to IdentityMapping Jonatan Werpers <jonatan@werpers.com> parents: 433 diff changeset	154 struct IdentityMapping{T,D} <: TensorMapping{T,D,D}
429 46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	155 size::NTuple{D,Int}
46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	156 end
434 648a36ebac99 Change name from LazyIdentity to IdentityMapping Jonatan Werpers <jonatan@werpers.com> parents: 433 diff changeset	157 export IdentityMapping
429 46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	158
434 648a36ebac99 Change name from LazyIdentity to IdentityMapping Jonatan Werpers <jonatan@werpers.com> parents: 433 diff changeset	159 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	160 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	161 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	162
434 648a36ebac99 Change name from LazyIdentity to IdentityMapping Jonatan Werpers <jonatan@werpers.com> parents: 433 diff changeset	163 range_size(tmi::IdentityMapping) = tmi.size
648a36ebac99 Change name from LazyIdentity to IdentityMapping Jonatan Werpers <jonatan@werpers.com> parents: 433 diff changeset	164 domain_size(tmi::IdentityMapping) = tmi.size
429 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 apply(tmi::IdentityMapping{T,D}, v::AbstractArray{T,D}, I::Vararg{Any,D}) where {T,D} = v[I...]
648a36ebac99 Change name from LazyIdentity to IdentityMapping Jonatan Werpers <jonatan@werpers.com> parents: 433 diff changeset	167 apply_transpose(tmi::IdentityMapping{T,D}, v::AbstractArray{T,D}, I::Vararg{Any,D}) where {T,D} = v[I...]
429 46acb2560451 Start implementing LazyIdentity Jonatan Werpers <jonatan@werpers.com> parents: 411 diff changeset	168
479 95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 465 diff changeset	169 """
95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 465 diff changeset	170 Base.:∘(tm, tmi)
95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 465 diff changeset	171 Base.:∘(tmi, tm)
95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 465 diff changeset	172
480 c1a366331e75 Fix typo in docs Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 479 diff changeset	173 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	174 """
95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 465 diff changeset	175 @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	176 @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	177 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	178 end
95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 465 diff changeset	179
95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 465 diff changeset	180 @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	181 @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	182 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	183 end
95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 465 diff changeset	184 # 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	185 @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	186 @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	187 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	188 end
95f3b9036801 Specialize composition operator for composing a tensormapping with an identitymapping. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 465 diff changeset	189
460 acffa3a1395d Move part of the documentation to the constructor Jonatan Werpers <jonatan@werpers.com> parents: 459 diff changeset	190
459 11e58c49fb46 Add some documentation Jonatan Werpers <jonatan@werpers.com> parents: 458 diff changeset	191 """
11e58c49fb46 Add some documentation Jonatan Werpers <jonatan@werpers.com> parents: 458 diff changeset	192 InflatedTensorMapping{T,R,D} <: TensorMapping{T,R,D}
11e58c49fb46 Add some documentation Jonatan Werpers <jonatan@werpers.com> parents: 458 diff changeset	193
11e58c49fb46 Add some documentation Jonatan Werpers <jonatan@werpers.com> parents: 458 diff changeset	194 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	195 """
449 14d60de71b72 Change to concrete fields in the type Jonatan Werpers <jonatan@werpers.com> parents: 448 diff changeset	196 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	197 before::IdentityMapping{T,D_before}
449 14d60de71b72 Change to concrete fields in the type Jonatan Werpers <jonatan@werpers.com> parents: 448 diff changeset	198 tm::TM
448 912ae510dec9 Use new name of IdentityMapping Jonatan Werpers <jonatan@werpers.com> parents: 447 diff changeset	199 after::IdentityMapping{T,D_after}
446 904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	200
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	201 function InflatedTensorMapping(before, tm::TensorMapping{T}, after) where T
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	202 R_before = range_dim(before)
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	203 R_middle = range_dim(tm)
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	204 R_after = range_dim(after)
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	205 R = R_before+R_middle+R_after
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	206
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	207 D_before = domain_dim(before)
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	208 D_middle = domain_dim(tm)
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	209 D_after = domain_dim(after)
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	210 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	211 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	212 end
446 904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	213 end
451 6cf234eef780 Add tests for constructors and size calculations Jonatan Werpers <jonatan@werpers.com> parents: 450 diff changeset	214 export InflatedTensorMapping
461 a0e40d16ba0e Better documentation Jonatan Werpers <jonatan@werpers.com> parents: 460 diff changeset	215 """
464 73882867c121 Align documentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 463 diff changeset	216 InflatedTensorMapping(before, tm, after)
465 f270d82fc9ad Change documention in an attempt to follow the guidelines Jonatan Werpers <jonatan@werpers.com> parents: 464 diff changeset	217 InflatedTensorMapping(before,tm)
f270d82fc9ad Change documention in an attempt to follow the guidelines Jonatan Werpers <jonatan@werpers.com> parents: 464 diff changeset	218 InflatedTensorMapping(tm,after)
461 a0e40d16ba0e Better documentation Jonatan Werpers <jonatan@werpers.com> parents: 460 diff changeset	219
a0e40d16ba0e Better documentation Jonatan Werpers <jonatan@werpers.com> parents: 460 diff changeset	220 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	221
f270d82fc9ad Change documention in an attempt to follow the guidelines Jonatan Werpers <jonatan@werpers.com> parents: 464 diff changeset	222 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	223
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 487 diff changeset	224 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	225 creating a nested `InflatedTensorMapping`.
461 a0e40d16ba0e Better documentation Jonatan Werpers <jonatan@werpers.com> parents: 460 diff changeset	226 """
464 73882867c121 Align documentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 463 diff changeset	227 InflatedTensorMapping(::IdentityMapping, ::TensorMapping, ::IdentityMapping)
493 df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 487 diff changeset	228
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 487 diff changeset	229 function InflatedTensorMapping(before, itm::InflatedTensorMapping, after)
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 487 diff changeset	230 return InflatedTensorMapping(
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 487 diff changeset	231 IdentityMapping(before.size..., itm.before.size...),
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 487 diff changeset	232 itm.tm,
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 487 diff changeset	233 IdentityMapping(itm.after.size..., after.size...),
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 487 diff changeset	234 )
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 487 diff changeset	235 end
df566372bb4f Implement constructors to avoid creating nested InflatedTensorMappings Jonatan Werpers <jonatan@werpers.com> parents: 487 diff changeset	236
462 c364e2908c6e Implement InflatedTensorMapping where one of before and after are missing. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 458 diff changeset	237 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	238 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	239 # 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	240 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	241
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	242 # 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	243
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	244 function range_size(itm::InflatedTensorMapping)
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	245 return flatten_tuple(
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	246 range_size(itm.before),
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	247 range_size(itm.tm),
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	248 range_size(itm.after),
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	249 )
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	250 end
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 function domain_size(itm::InflatedTensorMapping)
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	253 return flatten_tuple(
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	254 domain_size(itm.before),
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	255 domain_size(itm.tm),
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	256 domain_size(itm.after),
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	257 )
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	258 end
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	259
452 aeda2698166d Add tullio as a test dependency and add a test for apply Jonatan Werpers <jonatan@werpers.com> parents: 451 diff changeset	260 function apply(itm::InflatedTensorMapping{T,R,D}, v::AbstractArray{T,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	261 dim_before = range_dim(itm.before)
848dec405332 Change variable names in apply and apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 537 diff changeset	262 dim_domain = domain_dim(itm.tm)
848dec405332 Change variable names in apply and apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 537 diff changeset	263 dim_range = range_dim(itm.tm)
848dec405332 Change variable names in apply and apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 537 diff changeset	264 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	265
538 848dec405332 Change variable names in apply and apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 537 diff changeset	266 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	267
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	268 v_inner = view(v, view_index...)
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	269 return apply(itm.tm, v_inner, inner_index...)
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	270 end
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	271
526 be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 525 diff changeset	272 function apply_transpose(itm::InflatedTensorMapping{T,R,D}, v::AbstractArray{T,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	273 dim_before = range_dim(itm.before)
848dec405332 Change variable names in apply and apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 537 diff changeset	274 dim_domain = domain_dim(itm.tm)
848dec405332 Change variable names in apply and apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 537 diff changeset	275 dim_range = range_dim(itm.tm)
848dec405332 Change variable names in apply and apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 537 diff changeset	276 dim_after = range_dim(itm.after)
526 be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 525 diff changeset	277
538 848dec405332 Change variable names in apply and apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 537 diff changeset	278 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	279
be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 525 diff changeset	280 v_inner = view(v, view_index...)
be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 525 diff changeset	281 return apply_transpose(itm.tm, v_inner, inner_index...)
be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 525 diff changeset	282 end
be152486d136 Implement apply_transpose with tests Jonatan Werpers <jonatan@werpers.com> parents: 525 diff changeset	283
446 904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	284
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	285 """
533 aac7cc1fa79a Try to improve the naming in split_index() Jonatan Werpers <jonatan@werpers.com> parents: 532 diff changeset	286 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	287
533 aac7cc1fa79a Try to improve the naming in split_index() Jonatan Werpers <jonatan@werpers.com> parents: 532 diff changeset	288 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	289 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	290 Eg.
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	291 ```
537 41e82a5d4d48 Docs touch up Jonatan Werpers <jonatan@werpers.com> parents: 533 diff changeset	292 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	293 ```
520 fe86ac896377 Start refactoring split index and apply to accomodate future addition of apply_transpose Jonatan Werpers <jonatan@werpers.com> parents: 501 diff changeset	294
533 aac7cc1fa79a Try to improve the naming in split_index() Jonatan Werpers <jonatan@werpers.com> parents: 532 diff changeset	295 `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	296 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	297 `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	298
533 aac7cc1fa79a Try to improve the naming in split_index() Jonatan Werpers <jonatan@werpers.com> parents: 532 diff changeset	299 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	300
531 36dfc57e8e0b Write out some more documentation for split_index Jonatan Werpers <jonatan@werpers.com> parents: 530 diff changeset	301 The returned values satisfy
533 aac7cc1fa79a Try to improve the naming in split_index() Jonatan Werpers <jonatan@werpers.com> parents: 532 diff changeset	302 * `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	303 * `length(I_middle) == dim_index`
446 904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	304 """
533 aac7cc1fa79a Try to improve the naming in split_index() Jonatan Werpers <jonatan@werpers.com> parents: 532 diff changeset	305 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	306 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	307
533 aac7cc1fa79a Try to improve the naming in split_index() Jonatan Werpers <jonatan@werpers.com> parents: 532 diff changeset	308 view_index = (I_before..., ntuple((i)->:, dim_view)..., I_after...)
446 904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	309
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 return view_index, I_middle
446 904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	311 end
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	312
456 8f4c31e06689 Add docs and tests for slice_tuple Jonatan Werpers <jonatan@werpers.com> parents: 455 diff changeset	313 # 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	314 # See:
8f4c31e06689 Add docs and tests for slice_tuple Jonatan Werpers <jonatan@werpers.com> parents: 455 diff changeset	315 # https://github.com/JuliaLang/julia/issues/34884
8f4c31e06689 Add docs and tests for slice_tuple Jonatan Werpers <jonatan@werpers.com> parents: 455 diff changeset	316 # https://github.com/JuliaLang/julia/issues/30386
8f4c31e06689 Add docs and tests for slice_tuple Jonatan Werpers <jonatan@werpers.com> parents: 455 diff changeset	317 """
8f4c31e06689 Add docs and tests for slice_tuple Jonatan Werpers <jonatan@werpers.com> parents: 455 diff changeset	318 slice_tuple(t, Val(l), Val(u))
8f4c31e06689 Add docs and tests for slice_tuple Jonatan Werpers <jonatan@werpers.com> parents: 455 diff changeset	319
8f4c31e06689 Add docs and tests for slice_tuple Jonatan Werpers <jonatan@werpers.com> parents: 455 diff changeset	320 Get a slice of a tuple in a type stable way.
8f4c31e06689 Add docs and tests for slice_tuple Jonatan Werpers <jonatan@werpers.com> parents: 455 diff changeset	321 Equivalent to t[l:u] but type stable.
8f4c31e06689 Add docs and tests for slice_tuple Jonatan Werpers <jonatan@werpers.com> parents: 455 diff changeset	322 """
455 b86312d14873 Make split_index type stable Jonatan Werpers <jonatan@werpers.com> parents: 452 diff changeset	323 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	324 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	325 end
b86312d14873 Make split_index type stable Jonatan Werpers <jonatan@werpers.com> parents: 452 diff changeset	326
457 8fb6a5611c7a Add tests and docs for flatten_tuple Jonatan Werpers <jonatan@werpers.com> parents: 456 diff changeset	327 """
532 588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	328 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	329
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	330 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	331 E.g
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	332 ```
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	333 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	334 ```
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	335 """
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	336 function split_tuple(t::NTuple{N},::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	337 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	338 end
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 """
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	341 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	342
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	343 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	344 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	345 """
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	346 function split_tuple(t::NTuple{N},::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	347 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	348 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	349 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	350 end
588a843907de Add a split_tuple function to make split_index more readable Jonatan Werpers <jonatan@werpers.com> parents: 531 diff changeset	351
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 """
457 8fb6a5611c7a Add tests and docs for flatten_tuple Jonatan Werpers <jonatan@werpers.com> parents: 456 diff changeset	354 flatten_tuple(t)
8fb6a5611c7a Add tests and docs for flatten_tuple Jonatan Werpers <jonatan@werpers.com> parents: 456 diff changeset	355
8fb6a5611c7a Add tests and docs for flatten_tuple Jonatan Werpers <jonatan@werpers.com> parents: 456 diff changeset	356 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	357 """
446 904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	358 flatten_tuple(t::NTuple{N, Number} where N) = t
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	359 flatten_tuple(t::Tuple) = ((flatten_tuple.(t)...)...,) # simplify?
904aae1899df Start implementing InflatedTensorMapping Jonatan Werpers <jonatan@werpers.com> parents: 429 diff changeset	360 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	361
468 a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	362 """
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	363 LazyOuterProduct(tms...)
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	364
501 4b9d124fe984 Fix typo in documentation Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 499 diff changeset	365 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	366 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	367
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	368 First let
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	369 ```math
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	370 A = A_{I,J}
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	371 B = B_{M,N}
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	372 C = C_{P,Q}
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	373 ```
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	374
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	375 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	376
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	377 We use ``⊗`` to denote the outer product
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	378 ```math
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	379 (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	380 ```
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	381
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	382 We note that
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	383 ```math
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	384 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	385 ```
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	386 And that
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	387 ```math
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	388 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	389 ```
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	390 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
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	391 To apply ``A⊗B⊗C`` we evaluate
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 (A⊗B⊗C)v = [(A⊗I_{\|M\|}⊗I_{\|P\|}) [(I_{\|J\|}⊗B⊗I_{\|P\|}) [(I_{\|J\|}⊗I_{\|N\|}⊗C)v]]]
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	394 """
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	395 function LazyOuterProduct end
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	396 export LazyOuterProduct
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	397
470 0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	398 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	399 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	400 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	401
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	402 return itm1∘itm2
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	403 end
a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	404
491 2dc2eac27f75 Add special methods for Identity mappings Jonatan Werpers <jonatan@werpers.com> parents: 475 diff changeset	405 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	406 LazyOuterProduct(t1::TensorMapping, t2::IdentityMapping) = InflatedTensorMapping(t1, t2)
2dc2eac27f75 Add special methods for Identity mappings Jonatan Werpers <jonatan@werpers.com> parents: 475 diff changeset	407 LazyOuterProduct(t1::IdentityMapping, t2::TensorMapping) = InflatedTensorMapping(t1, t2)
2dc2eac27f75 Add special methods for Identity mappings Jonatan Werpers <jonatan@werpers.com> parents: 475 diff changeset	408
470 0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	409 LazyOuterProduct(tms::Vararg{TensorMapping}) = foldl(LazyOuterProduct, tms)
468 a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	410
474 3ad327378b2d Simplify implementation of \otimes Jonatan Werpers <jonatan@werpers.com> parents: 470 diff changeset	411 ⊗(a::TensorMapping, b::TensorMapping) = LazyOuterProduct(a,b)
470 0c3decc04649 Add some tests and fix some bugs Jonatan Werpers <jonatan@werpers.com> parents: 469 diff changeset	412 export ⊗
468 a52f38e72258 Start implementing function for outer products Jonatan Werpers <jonatan@werpers.com> parents: 461 diff changeset	413
498 5a600ec40ccc Merge in default Jonatan Werpers <jonatan@werpers.com> parents: 491 493 diff changeset	414
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	415 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	416 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	417 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	418 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	419 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	420
6a6b7eaf9edf Move exception handling to end of file and update error message. Vidar Stiernström <vidar.stiernstrom@it.uu.se> parents: 485 diff changeset	421 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	422 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	423 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	424 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	425 export 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	426
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 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	428 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	429 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	430 end

Mercurial > repos > public > sbplib_julia

annotate src/LazyTensors/lazy_tensor_operations.jl @ 541:62d96e2cd165 refactor/tensor_index_coupling