arXiv · 2206.11128
tntorch: Tensor Network Learning with PyTorch
Abstract
We present tntorch, a tensor learning framework that supports multiple decompositions (including Candecomp/Parafac, Tucker, and Tensor Train) under a unified interface. With our library, the user can learn and handle low-rank tensors with automatic differentiation, seamless GPU support, and the convenience of PyTorch's API. Besides decomposition algorithms, tntorch implements differentiable tensor algebra, rank truncation, cross-approximation, batch processing, comprehensive tensor arithmetics, and more.
Explore related subjects
Keep this discovery
Mikhail Usvyatsov, Rafael Ballester-Ripoll, Konrad Schindler. 2022-06-22. tntorch: Tensor Network Learning with PyTorch. https://arxiv.org/abs/2206.11128
Cite the original work for its findings. Save a collection to share your selection of sources.