class_docstr: Find the k nearest neighbors
class_docstr_tok: ["Find","the","k","nearest","neighbors"]
code: def __init__(self, n_neighbor):
      super(KNearNeighbors, 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 KNearNeighbors object from inherited base classes: nn with example: obj = KNearNeighbors(n_neighbor)