class_docstr: The Transition Down Module
class_docstr_tok: ["The","Transition","Down","Module"]
code: def __init__(self, n_points, batch_size, mlp_sizes, n_neighbors=64):
      super(TransitionDown, self).__init__()
      self.n_points = n_points
      self.frnn_graph = KNNGraphBuilder(n_neighbors)
      self.message = RelativePositionMessage(n_neighbors)
      self.conv = KNNConv(mlp_sizes, batch_size)
      self.batch_size = batch_size
  
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 TransitionDown object from inherited base classes: nn with example: obj = TransitionDown(n_points, batch_size, mlp_sizes, n_neighbors)