1. tntorch: Tensor Network Learning with PyTorch.
- Author
-
Usvyatsov, Mikhail, Ballester-Ripoll, Rafael, and Schindler, Konrad
- Subjects
- *
TENSOR algebra , *AUTOMATIC differentiation , *BATCH processing , *MULTILINEAR algebra , *ARITHMETIC , *APPLICATION program interfaces - 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, crossapproximation, batch processing, comprehensive tensor arithmetics, and more. [ABSTRACT FROM AUTHOR]
- Published
- 2022