Multi-head attention implemented in PyTorch
Project description
PyTorch Multi-Head Attention
Install
pip install torch-multi-head-attention
Usage
from torch_multi_head_attention import MultiHeadAttention
MultiHeadAttention(in_features=768, head_num=12)
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Close
Hashes for torch-multi-head-attention-0.15.1.tar.gz
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | e181602fe1ef6da8322cb6bc1ffb41f52d3658c54e3937040e8f186754bb3056 |
|
| MD5 | e24c9e56e808eee69921d26768f8bcca |
|
| BLAKE2b-256 | 8d73b0734654ec4c950270d32c3d4ffb7460e63df229021a52386bf86356e815 |
