tntorch：基于PyTorch的张量网络学习库
机器学习
2022-09-22 v2 数学软件
摘要
我们提出tntorch，一个张量学习框架，在统一接口下支持多种分解(包括Candecomp/Parafac、Tucker和Tensor Train)。借助我们的库，用户可以利用自动微分、无缝GPU支持和PyTorch API的便利性来学习并处理低秩张量。除分解算法外，tntorch还实现了可微张量代数、秩截断、交叉近似、批处理、全面的张量算术等。
引用
@article{arxiv.2206.11128,
title = {tntorch: Tensor Network Learning with PyTorch},
author = {Mikhail Usvyatsov and Rafael Ballester-Ripoll and Konrad Schindler},
journal= {arXiv preprint arXiv:2206.11128},
year = {2022}
}