mlsanityData to 14 Oct 2026

tntorch: Tensor Network Learning with PyTorch

arXiv 2206.11128 ↗First posted 22 Jun 2022cs.LG cs.MScited by 7

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

  1. 2015Tensorizing Neural Networkscited by 250
  2. 2016TensorLy: Tensor Learning in Pythoncited by 104
  3. 2019A Tensorized Transformer for Language Modelingcited by 86
  4. 2021Tensor Methods in Computer Vision and Deep Learningcited by 52
  5. 2019Factorized Higher-Order CNNs with an Application to Spatio-Temporal Emotion Estimationcited by 40
  6. 2017Scalable Gaussian Processes with Billions of Inducing Inputs via Tensor Train Decompositioncited by 28
  7. 2019Tensorized Embedding Layers for Efficient Model Compressioncited by 27
  8. 2019TensorNetwork: A Library for Physics and Machine Learningcited by 21
  9. 2022TTOpt: A Maximum Volume Quantized Tensor Train-based Optimization and its Application to Reinforcement Learningcited by 15
  10. 2020Hyperspectral Super-Resolution via Coupled Tensor Ring Factorizationcited by 12
  11. 2020Kronecker CP Decomposition with Fast Multiplication for Compressing RNNscited by 12
  12. 2019A Prior of a Googol Gaussians: a Tensor Ring Induced Prior for Generative Modelscited by 9
3 more
  1. 2021TedNet: A Pytorch Toolkit for Tensor Decomposition Networkscited by 9
  2. 2019MUSCO: Multi-Stage Compression of neural networkscited by 5
  3. 2021Cherry-Picking Gradients: Learning Low-Rank Embeddings of Visual Data via Differentiable Cross-Approximationcited by 4

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2022: 2; 2023: 1; 2024: 1; 2025: 2; 2026: 1.

7 papers.