Poster
in
Workshop: Second Workshop on Quantum Tensor Networks in Machine Learning

High Performance Hierarchical Tucker Tensor Learning Using GPU Tensor Cores

hao huang ⋅ Xiao-Yang Liu ⋅ Weiqin Tong ⋅ Tao Zhang ⋅ Anwar Walid

Abstract

We present efficient HT tensor learning primitives using GPU tensor cores and demonstrate three applications,namely third-order and high-order HT decompositions, a HT tensor layer for deep neural network and a HT tensor-tree structure quantum machine learning algorithm TTN. Experimental results demonstrated that our optimized algorithms have higher efficiency.

Chat is not available.