tntorch: Tensor Network Learning with PyTorch
Not on the map. Why?
Abstract (arXiv)
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.
Key facts. Cited by 7 papers in the map (data through 14 Oct 2026). Builds on 15 of its 29 references that are in the map. First posted on arXiv 22 Jun 2022.
Builds on
15 of 29 references.
References · 15
- 2015Tensorizing Neural Networkscited by 250
- 2016TensorLy: Tensor Learning in Pythoncited by 104
- 2019A Tensorized Transformer for Language Modelingcited by 86
- 2021Tensor Methods in Computer Vision and Deep Learningcited by 52
- 2019Factorized Higher-Order CNNs with an Application to Spatio-Temporal Emotion Estimationcited by 40
- 2017Scalable Gaussian Processes with Billions of Inducing Inputs via Tensor Train Decompositioncited by 28
- 2019Tensorized Embedding Layers for Efficient Model Compressioncited by 27
- 2019TensorNetwork: A Library for Physics and Machine Learningcited by 21
- 2022TTOpt: A Maximum Volume Quantized Tensor Train-based Optimization and its Application to Reinforcement Learningcited by 15
- 2020Hyperspectral Super-Resolution via Coupled Tensor Ring Factorizationcited by 12
- 2020Kronecker CP Decomposition with Fast Multiplication for Compressing RNNscited by 12
- 2019A Prior of a Googol Gaussians: a Tensor Ring Induced Prior for Generative Modelscited by 9
Built on by
Citing papers per year: 2022: 2; 2023: 1; 2024: 1; 2025: 2; 2026: 1.
7 papers.
- 2023Tensor Networks Meet Neural Networks: A Survey and Future Perspectivescited by 32
- 2022TT-NF: Tensor Train Neural Fieldscited by 7
- 2025tn4ml: Tensor Network Training and Customization for Machine Learningcited by 5
- 2025Tensorization of neural networks for improved privacy and interpretabilitycited by 3
- 2022T4DT: Tensorizing Time for Learning Temporal 3D Visual Datacited by 2
- 2024Leveraging Digital Perceptual Technologies for Remote Perception and Analysis of Human Biomechanical Processes: A Contactless Approach for Workload and Joint Force Assessmentcited by 0
- 2026Quantum-inspired tensor networks in machine learning modelscited by 0
- 2026Quantum-inspired tensor networks in machine learning models15 Apr 2026
- 2025tn4ml: Tensor Network Training and Customization for Machine Learning18 Feb 2025
- 2025Tensorization of neural networks for improved privacy and interpretability10 Jan 2025
- 2024Leveraging Digital Perceptual Technologies for Remote Perception and Analysis of Human Biomechanical Processes: A Contactless Approach for Workload and Joint Force Assessment2 Apr 2024
- 2023Tensor Networks Meet Neural Networks: A Survey and Future Perspectives22 Jan 2023
- 2022TT-NF: Tensor Train Neural Fields30 Sep 2022
- 2022T4DT: Tensorizing Time for Learning Temporal 3D Visual Data2 Aug 2022