Avoid NaN in feature normalization. (#2186)

This commit is contained in:
Fangjun Kuang
2025-05-08 11:22:47 +08:00
committed by GitHub
parent f00066db88
commit f9c99032c3

View File

@@ -49,6 +49,11 @@ static void ComputeMeanAndInvStd(const float *p, int32_t num_rows,
(*mean)[i] = t;
float stddev = std::sqrt(sum_sq[i] / num_rows - t * t);
if (stddev != stddev) {
stddev = 0;
}
(*inv_stddev)[i] = 1.0f / (stddev + 1e-5f);
}
}