class MultiheadAttention()

class MultiheadAttention(nn.Module):

Multi-headed attention.