code: def extra_repr(self) -> str:
      strs = []
      for p in [
          self.genpool_w1_head,
          self.genpool_b1_head,
          self.genpool_w2_head,
          self.genpool_b2_head,
      ]:
          strs.append(f"pool linear {p.shape}")
      return "\n".join(strs)
  
imports: from collections import OrderedDict
  import os
  import numpy
  import torch
  from torch import nn
  from gluoncv.torch.utils.coot_utils import truncated_normal_fill
label_desc: Use image node_id 2 for calling the AtnPool obj's underlying member method code with example usage: obj.extra_repr() and returns: str