Closed gau820827 closed 6 years ago
Rewrite the for loop as tensor manipulation to improve the performance of the attention module.
Rewrite the for loop as tensor manipulation to improve the performance of the attention module.