TedNet：一个用于张量分解网络的PyTorch工具包
机器学习
2021-11-23 v2 人工智能
摘要
张量分解网络（TDNs）因其固有的紧凑架构而流行。为给更多研究者提供灵活利用TDNs的途径，我们提出一个名为TedNet的PyTorch工具包。TedNet在传统深度神经网络层、卷积层与全连接层上实现了5种张量分解（即CANDECOMP/PARAFAC（CP）、块项Tucker（BTT）、Tucker-2、张量列（TT）与张量环（TR））。利用这些基础层，可简便地构建多种TDNs。TedNet可在https://github.com/tnbar/tednet获取。
引用
@article{arxiv.2104.05018,
title = {TedNet: A Pytorch Toolkit for Tensor Decomposition Networks},
author = {Yu Pan and Maolin Wang and Zenglin Xu},
journal= {arXiv preprint arXiv:2104.05018},
year = {2021}
}
备注
Accepted in Neurocomputing