Yes, this seems to be for Euclidean distance (I first wrote this code a while ago so it takes some head-scratching to figure it out again!).
The problem with the previous way of scaling is that SLIC-zero needs to know `zyx_wt` inside the loop to normalize the spatial distance.