Build KNN graph over some subset of Node features

Question

I have a point cloud that I want to use a graph neural network on. Each point in the point cloud is characterised by its positional coordinates as well as it\'s color. So a