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.