sakuranew/attention-pytorch

pytorch实现的基于attention is all your need提出的Q,K,V的attention模板和派生的attention实现。

Python