Mercurial > repos > public > sbplib_julia

annotate src/LazyTensors/LazyTensors.jl @ 1000:1091ac8c69ad refactor/lazy_tensors

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Rename another struct
author Jonatan Werpers <jonatan@werpers.com>
date Fri, 18 Mar 2022 21:54:01 +0100
parents 20cb83efb3f1
children a3df203861d3
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
184
6945c15a6a7a Rename package LazyTensor to LazyTensors
Jonatan Werpers <jonatan@werpers.com>
parents: 180
diff changeset
1 module LazyTensors
954
8f1b8827d389 Remove unused `using` of RegionIndices
Jonatan Werpers <jonatan@werpers.com>
parents: 335
diff changeset
2
995
1ba8a398af9c Rename types
Jonatan Werpers <jonatan@werpers.com>
parents: 992
diff changeset
3 export LazyTensorApplication
1ba8a398af9c Rename types
Jonatan Werpers <jonatan@werpers.com>
parents: 992
diff changeset
4 export LazyTensorTranspose
1ba8a398af9c Rename types
Jonatan Werpers <jonatan@werpers.com>
parents: 992
diff changeset
5 export LazyTensorComposition
976
089a1411dfc0 Move exports
Jonatan Werpers <jonatan@werpers.com>
parents: 954
diff changeset
6 export LazyLinearMap
995
1ba8a398af9c Rename types
Jonatan Werpers <jonatan@werpers.com>
parents: 992
diff changeset
7 export IdentityTensor
992
bc384aaade30 Add a bunch of todos and add a ScalingTensor
Jonatan Werpers <jonatan@werpers.com>
parents: 976
diff changeset
8 export ScalingTensor
995
1ba8a398af9c Rename types
Jonatan Werpers <jonatan@werpers.com>
parents: 992
diff changeset
9 export InflatedLazyTensor
976
089a1411dfc0 Move exports
Jonatan Werpers <jonatan@werpers.com>
parents: 954
diff changeset
10 export LazyOuterProduct
089a1411dfc0 Move exports
Jonatan Werpers <jonatan@werpers.com>
parents: 954
diff changeset
11 export ⊗
089a1411dfc0 Move exports
Jonatan Werpers <jonatan@werpers.com>
parents: 954
diff changeset
12 export SizeMismatch
089a1411dfc0 Move exports
Jonatan Werpers <jonatan@werpers.com>
parents: 954
diff changeset
13
996
aa72f067e771 Rename file
Jonatan Werpers <jonatan@werpers.com>
parents: 995
diff changeset
14 include("lazy_tensor.jl")
267
634453a4e1d8 Restructure code in LazyTensors
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 190
diff changeset
15 include("lazy_array.jl")
634453a4e1d8 Restructure code in LazyTensors
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 190
diff changeset
16 include("lazy_tensor_operations.jl")
997
20c376dffe84 Move tuple functions to their own file
Jonatan Werpers <jonatan@werpers.com>
parents: 996
diff changeset
17 include("tuple_manipulation.jl")
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
18
999
20cb83efb3f1 More operator definitions to top file
Jonatan Werpers <jonatan@werpers.com>
parents: 997
diff changeset
19 # Applying lazy tensors to vectors
20cb83efb3f1 More operator definitions to top file
Jonatan Werpers <jonatan@werpers.com>
parents: 997
diff changeset
20 Base.:*(a::LazyTensor, v::AbstractArray) = LazyTensorApplication(a,v)
20cb83efb3f1 More operator definitions to top file
Jonatan Werpers <jonatan@werpers.com>
parents: 997
diff changeset
21 Base.:*(a::LazyTensor, b::LazyTensor) = throw(MethodError(Base.:*,(a,b)))
20cb83efb3f1 More operator definitions to top file
Jonatan Werpers <jonatan@werpers.com>
parents: 997
diff changeset
22 Base.:*(a::LazyTensor, args::Union{LazyTensor, AbstractArray}...) = foldr(*,(a,args...))
20cb83efb3f1 More operator definitions to top file
Jonatan Werpers <jonatan@werpers.com>
parents: 997
diff changeset
23
20cb83efb3f1 More operator definitions to top file
Jonatan Werpers <jonatan@werpers.com>
parents: 997
diff changeset
24 # Addition and subtraction of lazy tensors
1000
1091ac8c69ad Rename another struct
Jonatan Werpers <jonatan@werpers.com>
parents: 999
diff changeset
25 Base.:+(tm1::LazyTensor{T,R,D}, tm2::LazyTensor{T,R,D}) where {T,R,D} = LazyTensorBinaryOperation{:+,T,R,D}(tm1,tm2)
1091ac8c69ad Rename another struct
Jonatan Werpers <jonatan@werpers.com>
parents: 999
diff changeset
26 Base.:-(tm1::LazyTensor{T,R,D}, tm2::LazyTensor{T,R,D}) where {T,R,D} = LazyTensorBinaryOperation{:-,T,R,D}(tm1,tm2)
999
20cb83efb3f1 More operator definitions to top file
Jonatan Werpers <jonatan@werpers.com>
parents: 997
diff changeset
27
20cb83efb3f1 More operator definitions to top file
Jonatan Werpers <jonatan@werpers.com>
parents: 997
diff changeset
28 # Composing lazy tensors
20cb83efb3f1 More operator definitions to top file
Jonatan Werpers <jonatan@werpers.com>
parents: 997
diff changeset
29 Base.:∘(s::LazyTensor, t::LazyTensor) = LazyTensorComposition(s,t)
20cb83efb3f1 More operator definitions to top file
Jonatan Werpers <jonatan@werpers.com>
parents: 997
diff changeset
30
20cb83efb3f1 More operator definitions to top file
Jonatan Werpers <jonatan@werpers.com>
parents: 997
diff changeset
31 # Outer products of tensors
20cb83efb3f1 More operator definitions to top file
Jonatan Werpers <jonatan@werpers.com>
parents: 997
diff changeset
32 ⊗(a::LazyTensor, b::LazyTensor) = LazyOuterProduct(a,b)
20cb83efb3f1 More operator definitions to top file
Jonatan Werpers <jonatan@werpers.com>
parents: 997
diff changeset
33
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
34 end # module