code: def __init__(self, in_ch, out_ch):
      super(AttenHead, self).__init__()
      # bottleneck channels for seg and attn heads
      bot_ch = 256
  
      self.atten_head = nn.Sequential(
          layers.ConvBNReLU(
              in_ch, bot_ch, 3, padding=1, bias_attr=False
          ),
          layers.ConvBNReLU(
              bot_ch, bot_ch, 3, padding=1, bias_attr=False
          ),
          nn.Conv2D(
              bot_ch, out_ch, kernel_size=(1, 1), bias_attr=False
          ),
          nn.Sigmoid(),
      )
  
imports: import math
  import paddle
  import paddle.nn
  from paddleseg.cvlibs import manager, param_init
  from paddleseg.utils import utils
  from paddleseg.models import layers
  from .ocrnet_nv import OCRNetNV
label_desc: Use image node_id 1 to create a new AttenHead object from inherited base classes: nn with example: obj = AttenHead(in_ch, out_ch)