Tensor Train Decomposition on TensorFlow (T3F)

JOURNAL OF MACHINE LEARNING RESEARCH(2020)

引用 62|浏览208
暂无评分
摘要
Tensor Train decomposition is used across many branches of machine learning. We present T3F a library for Tensor Train decomposition based on TensorFlow. T3F supports GPU execution, batch processing, automatic differentiation, and versatile functionality for the Riemannian optimization framework, which takes into account the underlying manifold structure to construct efficient optimization methods. The library makes it easier to implement machine learning papers that rely on the Tensor Train decomposition. T3F includes documentation, examples and 94% test coverage.
更多
查看译文
关键词
tensor decomposition,tensor train,software,gpu,tensorflow
AI 理解论文
溯源树
样例
生成溯源树,研究论文发展脉络
Chat Paper
正在生成论文摘要