search

gau820827 / AI-writer_Data2Doc

PyTorch Implementation of NBA game summary generator.
Apache License 2.0
83 stars 22 forks source link

Improve the efficiency of Attn module #32

Closed gau820827 closed 6 years ago

gau820827 commented 6 years ago

Rewrite the for loop as tensor manipulation to improve the performance of the attention module.