Avoid NaN in feature normalization. (#2186)

csukuangfj · web-flow · commit f9c99032c395 · 2025-05-08T11:22:47.000+08:00
diff --git a/sherpa-onnx/csrc/offline-stream.cc b/sherpa-onnx/csrc/offline-stream.cc
@@ -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);
   }
 }

Original file line number	Diff line number	Diff line change
`@@ -49,6 +49,11 @@ static void ComputeMeanAndInvStd(const float *p, int32_t num_rows,`
`49`	`49`	`(*mean)[i] = t;`
`50`	`50`
`51`	`51`	`float stddev = std::sqrt(sum_sq[i] / num_rows - t * t);`
	`52`	`+`
	`53`	`+ if (stddev != stddev) {`
	`54`	`+ stddev = 0;`
	`55`	`+ }`
	`56`	`+`
`52`	`57`	`(*inv_stddev)[i] = 1.0f / (stddev + 1e-5f);`
`53`	`58`	`}`
`54`	`59`	`}`