PyTorch Multi-Head Attention

February 16, 2019 ยท View on GitHub

Travis Coverage

Install

pip install torch-multi-head-attention

Usage

from torch_multi_head_attention import MultiHeadAttention

MultiHeadAttention(in_features=768, head_num=12)