pytorch
d9dedf05 - Adds support for 3D attn_mask in MultiHead Attention

Commit

1 year ago

Adds support for 3D attn_mask in MultiHead Attention

Author

LucasLLC

LucasLLC

Committer

LucasLLC

LucasLLC

Parents

Loading