Minor tweak to initializations

This commit is contained in:
Abe Levitan committed 2021-08-25 17:06:30 -04:00
1 parent 26148f80cb
commit ebafc44acb
2 files changed
+14 -7

No files matched your search

+1 -1
View File
@@ -633,7 +633,7 @@ def generate_subdominant_modes(dominant_mode, n_modes, circular=True):
dominant_fft = far_field(dominant_mode)
shape = dominant_mode.shape
center = ((shape[-2]-1)//2, (shape[-1]-1)//2)
center = ((shape[-2])//2, (shape[-1])//2)
i, j = np.mgrid[:shape[-2], :shape[-1]]
i = t.tensor(i - center[0]).to(dtype=dominant_fft.dtype,