class_docstr: A module which performs QKV attention. Matches legacy QKVAttention + input/ouput heads shaping
code: def __init__(self, n_heads):
      super().__init__()
      self.n_heads = n_heads
  
imports: import math
  from abc import abstractmethod
  import torch
  import torch.nn
  import torch.nn.functional
  from .fp16_util import convert_module_to_f16, convert_module_to_f32
  from .nn import avg_pool_nd, conv_nd, linear, normalization, timestep_embedding, zero_module
label_desc: Use image node_id 1 to create a new QKVAttention object from inherited base classes: nn with example: obj = QKVAttention(n_heads)