Mercurial > repos > public > sbplib_julia
view ext/SbplibSparseArrayKitExt.jl @ 1708:8787b2f9f5f8 feature/lazy_tensors/sparse_conversions
Add docstrings
author | Jonatan Werpers <jonatan@werpers.com> |
---|---|
date | Wed, 04 Sep 2024 22:47:41 +0200 |
parents | 3e9c3986930d |
children | 7379f492c4b3 |
line wrap: on
line source
module SbplibSparseArrayKitExt using Sbplib using Sbplib.LazyTensors using SparseArrayKit using Tokens """ SparseArray(t::LazyTensor) The sparse tensor representation of `t` with range dimensions to the left and domain dimensions to the right. If `L` is a `LazyTensor` with range and domain dimension 2 and `v` a 2-tensor, then `A = SparseArray(t)` is constructed so that `∑ₖ∑ₗA[i,j,k,l]*v[k,l] == L*v`§ """ function SparseArrayKit.SparseArray(t::LazyTensor) v = ArrayToken(:v, domain_size(t)...) return Tokens._to_tensor(t*v, range_size(t), domain_size(t)) end end