torchmil.nn.attention

• Attention Pool
• Sm Attention Pool
• Probabilistic Smooth Attention Pool
• Multihead Self-Attention
• Multihead Self-Attention with Relative Positional Encoding (iRPE)
• Nystrom Attention
• Multihead Cross-Attention