code: def __init__(self, in_channels, num_fiducial, loc_lr, model_name):
      super(TPS, self).__init__()
      self.loc_net = LocalizationNetwork(
          in_channels, num_fiducial, loc_lr, model_name
      )
      self.grid_generator = GridGenerator(
          self.loc_net.out_channels, num_fiducial
      )
      self.out_channels = in_channels
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import math
  import paddle
  from paddle import nn, ParamAttr
  from paddle.nn import functional
  import numpy
label_desc: Use image node_id 1 to create a new TPS object from inherited base classes: nn with example: obj = TPS(in_channels, num_fiducial, loc_lr, model_name)