Differentiable operations for simple shortest-distance

Shortestdistance is not differentiable due to som changes in API.