code: def __init__(self, dim1=1, dim2=2):
      super(SwapAxes, self).__init__()
      self.dim1 = dim1
      self.dim2 = dim2
  
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 SwapAxes object from inherited base classes: nn with example: obj = SwapAxes(dim1, dim2)