Shouldn't need to do anything extra, it's effectively already set to zero, as suggested in the parentetical. `diff_norm == 0` is iff `np.all(diff == 0, axis=1)` (the only way `diff_norm` is zero is when `diff` is all zeros).