Updating odak.learn.wave.get_impulse_response_fresnel_kernel().

kaanaksit · Sep 16, 2024 · 321760f · 321760f
1 parent dce8c49
commit 321760f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/odak/learn/wave/classical.py b/odak/learn/wave/classical.py
@@ -351,8 +351,8 @@ def get_impulse_response_fresnel_kernel(nu, nv, dx = 8e-6, wavelength = 515e-9,
     wxs = torch.linspace(- dx / 2., dx / 2., aperture_samples[0], device = device)
     wys = torch.linspace(- dx / 2., dx / 2., aperture_samples[1], device = device)
     h = torch.zeros(nu * scale, nv * scale, dtype = torch.complex64, device = device)
-    pxs = torch.linspace(- dx / 2., dx / 2., 5, device = device)
-    pys = torch.linspace(- dx / 2., dx / 2., 5, device = device)
+    pxs = torch.linspace(- dx / 2., dx / 2., aperture_samples[2], device = device)
+    pys = torch.linspace(- dx / 2., dx / 2., aperture_samples[3], device = device)
     for wx in tqdm(wxs):
         for wy in wys:
             for px in pxs: