knotgrass/attention

several types of attention modules written in PyTorch for learning purposes

Python