class_docstr: Multi-head Attention block from Transformer models.
code: def __init__(self, hidden_d, n_heads):
      super(MultiheadAttention, self).__init__()
      self.block = nn.MultiheadAttention(
          hidden_d, n_heads, batch_first=False
      )
  
imports: import numpy
  import torch
  import torch.nn
  from pycls.core.config import cfg
  from torch.nn import Module
label_desc: Use image node_id 1 to create a new MultiheadAttention object from inherited base classes: Module with example: obj = MultiheadAttention(hidden_d, n_heads)