code: def forward(self, x):
      params = self.local_net(x)
      x_transformed = F_affine3d(x, params.view(3, 4))
      return x_transformed
  
file_path: pywick/pywick/modules/stn.py
imports: import torch.nn
  from ..functions import F_affine2d, F_affine3d
label_desc: Use image node_id 2 for calling the STN3d obj's underlying member method code with example usage: obj.forward(x) and returns: x_transformed