Mercurial > repos > public > sbplib_julia
annotate ext/SbplibSparseArrayKitExt.jl @ 1802:b8cb38fd67ff feature/sbp_operators/laplace_curvilinear
Merge feature/grids/manifolds
author | Jonatan Werpers <jonatan@werpers.com> |
---|---|
date | Wed, 02 Oct 2024 08:53:01 +0200 |
parents | 3e9c3986930d |
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 |