torch-multi-head-attention icon indicating copy to clipboard operation
torch-multi-head-attention copied to clipboard

Published 20 hours ago • verified publisher icon CyberZHG
→

Metadata

Multi-head attention in PyTorch

PyTorch Multi-Head Attention

Travis Coverage

Install

pip install torch-multi-head-attention

Usage

from torch_multi_head_attention import MultiHeadAttention

MultiHeadAttention(in_features=768, head_num=12)

← Metadata

145
Stars
34
Forks
Watchers

Owner

verified publisher icon CyberZHG

Metadata

Multi-head attention in PyTorch

Back