softmax problem in contextDesc

[Yvaine]

Hi,
In contextDesc paper, I saw

in equation(5), the s is softmax(2-D), but the code in line170 of loss.py, list below
softmax_row = tf.nn.softmax(log_scale * dist_mat, axis=2)
the dist_mat has not subtract by 2, I don't know there is any wrong or i think wrong?

Thanks