rpi optimzation

misko · Mar 15, 2024 · 1f8e42a · 1f8e42a
1 parent d1e0fbc
commit 1f8e42a
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/spf/rf.py b/spf/rf.py
@@ -433,8 +433,7 @@ def beamformer_given_steering(
     steering_vectors,
     signal_matrix,
 ):
-    r = beamformer_given_steering_core(steering_vectors, signal_matrix)
-    return r.sum(axis=1) / r.shape[1]
+    return beamformer_given_steering_core(steering_vectors, signal_matrix).mean(axis=1)
 
 
 def beamformer_thetas(spacing):