K-Nearest Neighbor in Pytorch - View it on GitHub