Mercurial > repos > public > sbplib_julia
annotate ext/SbplibSparseArrayKitExt.jl @ 1699:3e9c3986930d feature/lazy_tensors/sparse_conversions
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
author | Jonatan Werpers <jonatan@werpers.com> |
---|---|
date | Mon, 02 Sep 2024 15:35:54 +0200 |
parents | |
children | 8787b2f9f5f8 |
rev | line source |
---|---|
1699
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
1 module SbplibSparseArrayKitExt |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
2 |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
3 using Sbplib |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
4 using Sbplib.LazyTensors |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
5 |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
6 using SparseArrayKit |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
7 using Tokens |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
8 |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
9 function SparseArrayKit.SparseArray(t::LazyTensor) |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
10 v = ArrayToken(:v, domain_size(t)...) |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
11 return Tokens._to_tensor(t*v, range_size(t), domain_size(t)) |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
12 end |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
13 |
3e9c3986930d
Add extensions for SparseArrays and SparseArrayKit that allow conversion of a LazyTensor
Jonatan Werpers <jonatan@werpers.com>
parents:
diff
changeset
|
14 end |