Mercurial > repos > public > sbplib_julia
annotate src/LazyTensors/LazyTensors.jl @ 1037:9e76bf19904c refactor/lazy_tensors
Drop more prefixes
author | Jonatan Werpers <jonatan@werpers.com> |
---|---|
date | Tue, 22 Mar 2022 14:10:53 +0100 |
parents | 6abbb2c6c3e4 |
children | 93f87d5d9fbb |
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 |
1037 | 4 export TensorTranspose |
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 |
1037 | 27 Base.:+(s::LazyTensor, t::LazyTensor) = ElementwiseTensorOperation{:+}(s,t) |
28 Base.:-(s::LazyTensor, t::LazyTensor) = ElementwiseTensorOperation{:-}(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 |