From 321760f2f2f3e2639301ecb32535cc801f53dd64 Mon Sep 17 00:00:00 2001 From: kaanaksit Date: Mon, 16 Sep 2024 13:29:17 +0100 Subject: [PATCH] Updating odak.learn.wave.get_impulse_response_fresnel_kernel(). --- odak/learn/wave/classical.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/odak/learn/wave/classical.py b/odak/learn/wave/classical.py index 3f8e3009..ab2bf631 100644 --- 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: