code: def __init__(self, k):
      self.k = k
  
imports: import torch
  from torch_geometric.nn import knn_interpolate, knn
  from torch_scatter import scatter_add
  from torch_geometric.data import Data
label_desc: Use image node_id 1 to create a new KNNInterpolate object  with example: obj = KNNInterpolate(k)