Neural Diffusion Distance for Image Segmentation