PyTorch Multi-Head Attention

Install

pip install torch-multi-head-attention

from torch_multi_head_attention import MultiHeadAttention

MultiHeadAttention(in_features=768, head_num=12)

Name		Name	Last commit message	Last commit date
Latest commit History 4 Commits
.github		.github
tests		tests
torch_multi_head_attention		torch_multi_head_attention
.gitignore		.gitignore
.travis.yml		.travis.yml
LICENSE		LICENSE
MANIFEST.in		MANIFEST.in
README.md		README.md
lint.sh		lint.sh
publish.sh		publish.sh
requirements-dev.txt		requirements-dev.txt
requirements.txt		requirements.txt
setup.py		setup.py
test.sh		test.sh