code: def __init__(self, local_net):
      super(STN2d, self).__init__()
      self.local_net = local_net
  
file_path: pywick/pywick/modules/stn.py
imports: import torch.nn
  from ..functions import F_affine2d, F_affine3d
label_desc: Use image node_id 1 to create a new STN2d object from inherited base classes: nn with example: obj = STN2d(local_net)