class_docstr: The Transition Up Module
class_docstr_tok: ["The","Transition","Up","Module"]
code: def __init__(self, dim1, dim2, dim_out):
      super(TransitionUp, self).__init__()
      self.fc1 = nn.Sequential(
          nn.Linear(dim1, dim_out),
          SwapAxes(),
          nn.BatchNorm1d(dim_out),  # TODO
          SwapAxes(),
          nn.ReLU(),
      )
      self.fc2 = nn.Sequential(
          nn.Linear(dim2, dim_out),
          SwapAxes(),
          nn.BatchNorm1d(dim_out),  # TODO
          SwapAxes(),
          nn.ReLU(),
      )
      self.fp = FeaturePropagation(-1, [])
  
imports: import dgl
  import torch
  import torch.nn
  import torch.nn.functional
  from dgl.geometry import farthest_point_sampler
label_desc: Use image node_id 1 to create a new TransitionUp object from inherited base classes: nn with example: obj = TransitionUp(dim1, dim2, dim_out)