protected against divide by zero issues with zero input signal

drowe67 · Jan 9, 2025 · 0530eea · 0530eea
1 parent 745686f
commit 0530eea
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/radae/dsp.py b/radae/dsp.py
@@ -403,7 +403,7 @@ def update_snr_est(self, rx_sym_pilots, rx_pilots):
       rx_phase = torch.angle(rx_pilots[0,:])
       Rcn_hat = Pcn_hat * torch.exp(-1j*rx_phase)
       S1 = torch.sum(torch.abs(Pcn_hat)**2)
-      S2 = torch.sum(torch.abs(Rcn_hat.imag)**2)   
+      S2 = torch.sum(torch.abs(Rcn_hat.imag)**2) + 1E-12 
       snr_est = S1/(2*S2) - 1
       # remove occasional illegal values
       if snr_est <= 0: