TensorLy
TensorLy is a Python library that aims at making tensor learning simple and accessible. It allows to easily perform tensor decomposition, tensor learning and tensor algebra. Its backend system allows to seamlessly perform computation with NumPy, PyTorch, JAX, TensorFlow, CuPy or Paddle, and run methods at scale on CPU or GPU.
Website: https://tensorly.org
Source-code: https://github.com/tensorly/tensorly
Jupyter Notebooks: https://github.com/JeanKossaifi/tensorly-notebooks
Installing TensorLy
The only pre-requisite is to have Python 3 installed. The easiest way is via the Anaconda distribution.
With pip (recommended) |
With conda |
|
|
Development (from git) |
|
|
|
Note: TensorLy depends on NumPy by default. If you want to use other backends, you will need to install these packages separately.
For detailed instruction, please see the documentation.
Quickstart
Creating tensors
Create a small third order tensor of size 3 x 4 x 2, from a NumPy array and perform simple operations on it:
import tensorly as tl
import numpy as np
tensor = tl.tensor(np.arange(24).reshape((3, 4, 2)), dtype=tl.float64)
unfolded = tl.unfold(tensor, mode=0)
tl.fold(unfolded, mode=0, shape=tensor.shape)You can also create random tensors:
from tensorly import random
# A random tensor
tensor = random.random_tensor((3, 4, 2))
# A random CP tensor in factorized form
cp_tensor = random.random_tensor(shape=(3, 4, 2), rank='same')You can also create tensors in TT-format, Tucker, etc, see random tensors.
Setting the backend
You can change the backend to perform computation with a different framework. By default, the backend is NumPy, but you can also perform the computation using PyTorch, TensorFlow, JAX, CuPy or Paddle (requires to have installed them first). For instance, after setting the backend to PyTorch, all the computation is done by PyTorch, and tensors can be created on GPU:
tl.set_backend('pytorch') # Or 'numpy', 'tensorflow', 'cupy' or 'jax'
tensor = tl.tensor(np.arange(24).reshape((3, 4, 2)), device='cuda:0')
type(tensor) # torch.TensorTensor decomposition
Applying tensor decomposition is easy:
from tensorly.decomposition import tucker
# Apply Tucker decomposition
tucker_tensor = tucker(tensor, rank=[2, 2, 2])
# Reconstruct the full tensor from the decomposed form
tl.tucker_to_tensor(tucker_tensor)We have many more decompositions available, be sure to check them out!
Next steps
This is just a very quick introduction to some of the basic features of TensorLy. For more information on getting started, checkout the user-guide and for a detailed reference of the functions and their documentation, refer to the API
If you see a bug, open an issue, or better yet, a pull-request!
Contributing code
All contributions are welcome! So if you have a cool tensor method you want to add, if you spot a bug or even a typo or mistake in the documentation, please report it, and even better, open a Pull-Request on GitHub.
Before you submit your changes, you should make sure your code adheres to our style-guide. The easiest way to do this is with black:
pip install black
black .Running the tests
Testing and documentation are an essential part of this package and all functions come with uni-tests and documentation.
The tests are ran using the pytest package. First install pytest:
pip install pytest
Then to run the test, simply run, in the terminal:
pytest -v tensorlyAlternatively, you can specify for which backend you wish to run the tests:
TENSORLY_BACKEND='numpy' pytest -v tensorlyCiting
If you use TensorLy in an academic paper, please cite [1]:
@article{tensorly,
author = {Jean Kossaifi and Yannis Panagakis and Anima Anandkumar and Maja Pantic},
title = {TensorLy: Tensor Learning in Python},
journal = {Journal of Machine Learning Research},
year = {2019},
volume = {20},
number = {26},
pages = {1-6},
url = {http://jmlr.org/papers/v20/18-277.html}
}
1 maintainer
Metadata
1 maintainer
Release files for tensorly 0.10.0
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| tensorly-0.10.0.tar.gz | 7.3 MB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| tensorly-0.10.0-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 14.8 MB
Release files / tensorly-0.10.0.tar.gz
| Download URL | tensorly-0.10.0.tar.gz |
|---|---|
| Size | 7.3 MB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
56a2c55062b285f4c39c6dc603e4883e02d9d5d2c318fd36f5610967a4425e9c
|
|
BLAKE2b-256 checksum How to use checksums |
c265b51b379b19f7472feedbe623522f80bcf44d8aa6ca206083c026112ced29
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/7.0.0 CPython/3.13.14
|
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on Oct 5, 2026.
Transparency logRelease files / tensorly-0.10.0-py3-none-any.whl
| Download URL | tensorly-0.10.0-py3-none-any.whl |
|---|---|
| Size | 7.4 MB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
90becb3d28023638ec60d19fbf8d87735b4ce1537fe3eb881897d69e107b882d
|
|
BLAKE2b-256 checksum How to use checksums |
17020269a1573b9f2ea6365e12b87debaf8b2cdad8e3fc4f84d9f4dc6be68adb
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/7.0.0 CPython/3.13.14
|
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on Oct 5, 2026.
Transparency log
