class_docstr: Compute the input feature from neighbors
class_docstr_tok: ["Compute","the","input","feature","from","neighbors"]
code: def __init__(self, n_neighbor):
      super(KNNMessage, self).__init__()
      self.n_neighbor = n_neighbor
  
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 KNNMessage object from inherited base classes: nn with example: obj = KNNMessage(n_neighbor)