Jun, 2022

使用PyTorch进行张量网络学习

TL;DRtntorch是一个支持多种分解(包括Candecomp/Parafac、Tucker、和Tensor Train)的张量学习框架,可以自动微分、无缝支持GPU、拥有PyTorch的API,不仅实现了分解算法,还实现了可微分的张量代数、秩截断、交叉逼近、批处理以及全面的张量算术等