Mercurial > repos > public > sbplib_julia

annotate src/LazyTensors/LazyTensors.jl @ 1897:ed50eec18365 refactor/lazy_tensors/elementwise_ops

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Change implementation of apply to use sum()
author Jonatan Werpers <jonatan@werpers.com>
date Fri, 31 Jan 2025 11:03:09 +0100
parents e1077273eda5
children
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
1767
dfb43fdac9fc Move exports to package top files
Jonatan Werpers <jonatan@werpers.com>
parents: 1300
diff changeset
3 export LazyTensor
dfb43fdac9fc Move exports to package top files
Jonatan Werpers <jonatan@werpers.com>
parents: 1300
diff changeset
4 export apply
dfb43fdac9fc Move exports to package top files
Jonatan Werpers <jonatan@werpers.com>
parents: 1300
diff changeset
5 export apply_transpose
dfb43fdac9fc Move exports to package top files
Jonatan Werpers <jonatan@werpers.com>
parents: 1300
diff changeset
6 export range_dim, domain_dim
dfb43fdac9fc Move exports to package top files
Jonatan Werpers <jonatan@werpers.com>
parents: 1300
diff changeset
7 export range_size, domain_size
dfb43fdac9fc Move exports to package top files
Jonatan Werpers <jonatan@werpers.com>
parents: 1300
diff changeset
8
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1006
diff changeset
9 export TensorApplication
1037
9e76bf19904c Drop more prefixes
Jonatan Werpers <jonatan@werpers.com>
parents: 1017
diff changeset
10 export TensorTranspose
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1006
diff changeset
11 export TensorComposition
1836
368999a2e243 Add TensorNegation
Jonatan Werpers <jonatan@werpers.com>
parents: 1835
diff changeset
12 export TensorNegation
1837
200971c71657 Refactor ElementwiseTensorOperation into TensorSum and use TensorNegation for handling subtraction
Jonatan Werpers <jonatan@werpers.com>
parents: 1836
diff changeset
13 export TensorSum
995
1ba8a398af9c Rename types
Jonatan Werpers <jonatan@werpers.com>
parents: 992
diff changeset
14 export IdentityTensor
992
bc384aaade30 Add a bunch of todos and add a ScalingTensor
Jonatan Werpers <jonatan@werpers.com>
parents: 976
diff changeset
15 export ScalingTensor
1071
93f87d5d9fbb Add a lazy diagonal tensor
Jonatan Werpers <jonatan@werpers.com>
parents: 1037
diff changeset
16 export DiagonalTensor
93f87d5d9fbb Add a lazy diagonal tensor
Jonatan Werpers <jonatan@werpers.com>
parents: 1037
diff changeset
17 export DenseTensor
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1006
diff changeset
18 export InflatedTensor
976
089a1411dfc0 Move exports
Jonatan Werpers <jonatan@werpers.com>
parents: 954
diff changeset
19 export LazyOuterProduct
089a1411dfc0 Move exports
Jonatan Werpers <jonatan@werpers.com>
parents: 954
diff changeset
20 export ⊗
1006
d9476fede83d Add check methods for range size
Jonatan Werpers <jonatan@werpers.com>
parents: 1002
diff changeset
21 export DomainSizeMismatch
d9476fede83d Add check methods for range size
Jonatan Werpers <jonatan@werpers.com>
parents: 1002
diff changeset
22 export RangeSizeMismatch
976
089a1411dfc0 Move exports
Jonatan Werpers <jonatan@werpers.com>
parents: 954
diff changeset
23
1767
dfb43fdac9fc Move exports to package top files
Jonatan Werpers <jonatan@werpers.com>
parents: 1300
diff changeset
24 export LazyArray
dfb43fdac9fc Move exports to package top files
Jonatan Werpers <jonatan@werpers.com>
parents: 1300
diff changeset
25 export LazyFunctionArray
dfb43fdac9fc Move exports to package top files
Jonatan Werpers <jonatan@werpers.com>
parents: 1300
diff changeset
26 export +̃, -̃, *̃, /̃
dfb43fdac9fc Move exports to package top files
Jonatan Werpers <jonatan@werpers.com>
parents: 1300
diff changeset
27
996
aa72f067e771 Rename file
Jonatan Werpers <jonatan@werpers.com>
parents: 995
diff changeset
28 include("lazy_tensor.jl")
1002
271aa6ae1055 Split out a file for tensor types
Jonatan Werpers <jonatan@werpers.com>
parents: 1001
diff changeset
29 include("tensor_types.jl")
267
634453a4e1d8 Restructure code in LazyTensors
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 190
diff changeset
30 include("lazy_array.jl")
634453a4e1d8 Restructure code in LazyTensors
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 190
diff changeset
31 include("lazy_tensor_operations.jl")
997
20c376dffe84 Move tuple functions to their own file
Jonatan Werpers <jonatan@werpers.com>
parents: 996
diff changeset
32 include("tuple_manipulation.jl")
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
33
999
20cb83efb3f1 More operator definitions to top file
Jonatan Werpers <jonatan@werpers.com>
parents: 997
diff changeset
34 # Applying lazy tensors to vectors
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1006
diff changeset
35 Base.:*(a::LazyTensor, v::AbstractArray) = TensorApplication(a,v)
999
20cb83efb3f1 More operator definitions to top file
Jonatan Werpers <jonatan@werpers.com>
parents: 997
diff changeset
36 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
37 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
38
20cb83efb3f1 More operator definitions to top file
Jonatan Werpers <jonatan@werpers.com>
parents: 997
diff changeset
39 # Addition and subtraction of lazy tensors
1837
200971c71657 Refactor ElementwiseTensorOperation into TensorSum and use TensorNegation for handling subtraction
Jonatan Werpers <jonatan@werpers.com>
parents: 1836
diff changeset
40 Base.:+(ts::LazyTensor...) = TensorSum(ts...)
1836
368999a2e243 Add TensorNegation
Jonatan Werpers <jonatan@werpers.com>
parents: 1835
diff changeset
41 Base.:-(t::LazyTensor) = TensorNegation(t)
1839
e1077273eda5 Separate the logic of how TensorSum works from the semantics of + and -
Jonatan Werpers <jonatan@werpers.com>
parents: 1837
diff changeset
42 Base.:-(s::LazyTensor, t::LazyTensor) = s + (-t)
e1077273eda5 Separate the logic of how TensorSum works from the semantics of + and -
Jonatan Werpers <jonatan@werpers.com>
parents: 1837
diff changeset
43 ## Specializations to flatten the nesting of tensors. This helps Julia during inference.
e1077273eda5 Separate the logic of how TensorSum works from the semantics of + and -
Jonatan Werpers <jonatan@werpers.com>
parents: 1837
diff changeset
44 Base.:+(t::TensorSum, s::TensorSum) = TensorSum(t.tms..., s.tms...)
e1077273eda5 Separate the logic of how TensorSum works from the semantics of + and -
Jonatan Werpers <jonatan@werpers.com>
parents: 1837
diff changeset
45 Base.:+(t::TensorSum, s::LazyTensor) = TensorSum(t.tms..., s)
e1077273eda5 Separate the logic of how TensorSum works from the semantics of + and -
Jonatan Werpers <jonatan@werpers.com>
parents: 1837
diff changeset
46 Base.:+(t::LazyTensor, s::TensorSum) = TensorSum(t, s.tms...)
999
20cb83efb3f1 More operator definitions to top file
Jonatan Werpers <jonatan@werpers.com>
parents: 997
diff changeset
47
20cb83efb3f1 More operator definitions to top file
Jonatan Werpers <jonatan@werpers.com>
parents: 997
diff changeset
48 # Composing lazy tensors
1017
6abbb2c6c3e4 Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents: 1006
diff changeset
49 Base.:∘(s::LazyTensor, t::LazyTensor) = TensorComposition(s,t)
1300
a922aa69eb83 Fix type inference of nested tensorcompositions by making the compositions right associative
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents: 1071
diff changeset
50 Base.:∘(s::TensorComposition, t::LazyTensor) = s.t1∘(s.t2∘t)
999
20cb83efb3f1 More operator definitions to top file
Jonatan Werpers <jonatan@werpers.com>
parents: 997
diff changeset
51
20cb83efb3f1 More operator definitions to top file
Jonatan Werpers <jonatan@werpers.com>
parents: 997
diff changeset
52 # Outer products of tensors
20cb83efb3f1 More operator definitions to top file
Jonatan Werpers <jonatan@werpers.com>
parents: 997
diff changeset
53 ⊗(a::LazyTensor, b::LazyTensor) = LazyOuterProduct(a,b)
20cb83efb3f1 More operator definitions to top file
Jonatan Werpers <jonatan@werpers.com>
parents: 997
diff changeset
54
180
b7397ae8afaf Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff changeset
55 end # module