tencat is a python package that offers functionality to slice sparse tensors
Project description
Tencat is a Python library that offers basic sparse tensor slicing capabilities.
Installation
Use the package manager pip to install tencat.
pip install tencat
Usage
Tencat implements the coordinate format in SparseTensor using a (N,d) ndarray as indices, a (N,) ndarray as values and an optional d length size tuple.
from tencat import tencat as tc
nnz = 100 # Number of nonzero elements
ndim = 4 # Number of dimensions
indices = np.randint(100,(nnz,ndim)) # Create random indices
values = np.random.rand(nnz,) # create random elements
dense_shape = np.amax(indices,axis=0) + np.random.randint(5,size=(ndim,))
sparse_tc = tc.SparseTensor(indices,values,dense_shape)
sliced_tc = sparse_tc.slice((1,5,6,4),(3,2,10,5))
License
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
tencat-0.3.tar.gz
(3.0 kB
view hashes)
Built Distribution
tencat-0.3-py3-none-any.whl
(4.0 kB
view hashes)