code: def __init__(
      self,
      pool_type,
      C_in,
      C_out,
      kernel_size,
      stride,
      padding,
      affine=False,
  ):
      super().__init__()
      self.preproc = StdConv(C_in, C_out)
      self.pool = Pool(pool_type, kernel_size, stride, padding)
      self.bn = nn.BatchNorm2d(C_out, affine=affine)
  
imports: import torch
  import torch.nn
label_desc: Use image node_id 1 to create a new PoolBranch object from inherited base classes: nn with example: obj = PoolBranch(pool_type, C_in, C_out, kernel_size, stride, padding, affine)