class_docstr: Multi-head Attention block from Transformer models.
code: def forward(self, query, key, value, need_weights=False):
      return self.block(
          query=query, key=key, value=value, need_weights=need_weights
      )
  
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 2 for calling the MultiheadAttention obj's underlying member method code with example usage: obj.forward(query, key, value, need_weights) and returns: self