We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo ScienceWISE logo

Computer Science > Machine Learning

Title: TedNet: A Pytorch Toolkit for Tensor Decomposition Networks

Abstract: Tensor Decomposition Networks (TDNs) prevail for their inherent compact architectures. To give more researchers a flexible way to exploit TDNs, we present a Pytorch toolkit named TedNet. TedNet implements 5 kinds of tensor decomposition(i.e., CANDECOMP/PARAFAC (CP), Block-Term Tucker (BTT), Tucker-2, Tensor Train (TT) and Tensor Ring (TR) on traditional deep neural layers, the convolutional layer and the fully-connected layer. By utilizing the basic layers, it is simple to construct a variety of TDNs. TedNet is available at this https URL
Comments: Accepted in Neurocomputing
Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI)
DOI: 10.1016/j.neucom.2021.10.064
Cite as: arXiv:2104.05018 [cs.LG]
  (or arXiv:2104.05018v2 [cs.LG] for this version)

Submission history

From: Yu Pan [view email]
[v1] Sun, 11 Apr 2021 14:04:06 GMT (313kb,D)
[v2] Sun, 21 Nov 2021 13:15:43 GMT (147kb,D)

Link back to: arXiv, form interface, contact.