Skip to yearly menu bar Skip to main content

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


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.