MultiheadAttention — PyTorch 2.4 documentation
pytorch.org