Mercurial > repos > public > sbplib_julia
annotate src/LazyTensors/LazyTensors.jl @ 1017:6abbb2c6c3e4 refactor/lazy_tensors
Remove the Lazy prefix on some types
author | Jonatan Werpers <jonatan@werpers.com> |
---|---|
date | Mon, 21 Mar 2022 15:22:22 +0100 |
parents | d9476fede83d |
children | 9e76bf19904c |
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 |
1017
6abbb2c6c3e4
Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents:
1006
diff
changeset
|
3 export TensorApplication |
995 | 4 export LazyTensorTranspose |
1017
6abbb2c6c3e4
Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents:
1006
diff
changeset
|
5 export TensorComposition |
6abbb2c6c3e4
Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents:
1006
diff
changeset
|
6 export DenseTensor |
995 | 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 |
1017
6abbb2c6c3e4
Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents:
1006
diff
changeset
|
9 export InflatedTensor |
976 | 10 export LazyOuterProduct |
11 export ⊗ | |
1006
d9476fede83d
Add check methods for range size
Jonatan Werpers <jonatan@werpers.com>
parents:
1002
diff
changeset
|
12 export DomainSizeMismatch |
d9476fede83d
Add check methods for range size
Jonatan Werpers <jonatan@werpers.com>
parents:
1002
diff
changeset
|
13 export RangeSizeMismatch |
976 | 14 |
996 | 15 include("lazy_tensor.jl") |
1002
271aa6ae1055
Split out a file for tensor types
Jonatan Werpers <jonatan@werpers.com>
parents:
1001
diff
changeset
|
16 include("tensor_types.jl") |
267
634453a4e1d8
Restructure code in LazyTensors
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
190
diff
changeset
|
17 include("lazy_array.jl") |
634453a4e1d8
Restructure code in LazyTensors
Vidar Stiernström <vidar.stiernstrom@it.uu.se>
parents:
190
diff
changeset
|
18 include("lazy_tensor_operations.jl") |
997
20c376dffe84
Move tuple functions to their own file
Jonatan Werpers <jonatan@werpers.com>
parents:
996
diff
changeset
|
19 include("tuple_manipulation.jl") |
180
b7397ae8afaf
Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
20 |
999
20cb83efb3f1
More operator definitions to top file
Jonatan Werpers <jonatan@werpers.com>
parents:
997
diff
changeset
|
21 # Applying lazy tensors to vectors |
1017
6abbb2c6c3e4
Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents:
1006
diff
changeset
|
22 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
|
23 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
|
24 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
|
25 |
20cb83efb3f1
More operator definitions to top file
Jonatan Werpers <jonatan@werpers.com>
parents:
997
diff
changeset
|
26 # Addition and subtraction of lazy tensors |
1001
a3df203861d3
Simplify some type parameter usage
Jonatan Werpers <jonatan@werpers.com>
parents:
1000
diff
changeset
|
27 Base.:+(s::LazyTensor, t::LazyTensor) = LazyTensorBinaryOperation{:+}(s,t) |
a3df203861d3
Simplify some type parameter usage
Jonatan Werpers <jonatan@werpers.com>
parents:
1000
diff
changeset
|
28 Base.:-(s::LazyTensor, t::LazyTensor) = LazyTensorBinaryOperation{:-}(s,t) |
999
20cb83efb3f1
More operator definitions to top file
Jonatan Werpers <jonatan@werpers.com>
parents:
997
diff
changeset
|
29 |
20cb83efb3f1
More operator definitions to top file
Jonatan Werpers <jonatan@werpers.com>
parents:
997
diff
changeset
|
30 # Composing lazy tensors |
1017
6abbb2c6c3e4
Remove the Lazy prefix on some types
Jonatan Werpers <jonatan@werpers.com>
parents:
1006
diff
changeset
|
31 Base.:∘(s::LazyTensor, t::LazyTensor) = TensorComposition(s,t) |
999
20cb83efb3f1
More operator definitions to top file
Jonatan Werpers <jonatan@werpers.com>
parents:
997
diff
changeset
|
32 |
20cb83efb3f1
More operator definitions to top file
Jonatan Werpers <jonatan@werpers.com>
parents:
997
diff
changeset
|
33 # Outer products of tensors |
20cb83efb3f1
More operator definitions to top file
Jonatan Werpers <jonatan@werpers.com>
parents:
997
diff
changeset
|
34 ⊗(a::LazyTensor, b::LazyTensor) = LazyOuterProduct(a,b) |
20cb83efb3f1
More operator definitions to top file
Jonatan Werpers <jonatan@werpers.com>
parents:
997
diff
changeset
|
35 |
180
b7397ae8afaf
Move tensor mappings to a package LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
36 end # module |