search training images for each test image in MNIST now