From 56daf05d005a13e228a00404ef224f15b7a46892 Mon Sep 17 00:00:00 2001 From: Joshua Mukherjee Date: Wed, 20 May 2026 16:11:27 +0100 Subject: [PATCH] type --- .../src/acoustools/Utilities/Piston_model_gradients.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/acoustools/src/acoustools/Utilities/Piston_model_gradients.py b/acoustools/src/acoustools/Utilities/Piston_model_gradients.py index 6ab5e46..7332f16 100644 --- a/acoustools/src/acoustools/Utilities/Piston_model_gradients.py +++ b/acoustools/src/acoustools/Utilities/Piston_model_gradients.py @@ -35,11 +35,13 @@ def compute_gradients(points, transducers = TRANSDUCERS, p_ref = Constants.P_ref points = points.expand((-1,M,-1,-1)) diff = points - transducers - distances = torch.sqrt(torch.sum(diff**2, 2)) - + diff = diff + 0j + distances = torch.sqrt(torch.sum(diff**2, 2)) + sin_theta = torch.norm(torch.cross(diff, transducer_norms_exp, dim=2),2, dim=2) / distances partialFpartialU = -1* (k**2 * transducer_radius**2)/4 * sin_theta + (k**4 * transducer_radius**4)/48 * sin_theta**3 + partialUpartiala = torch.sum(diff * transducer_norms_exp, dim=2) / distances #d/dx sin(x) = cos(x) partialUpartiala = partialUpartiala.unsqueeze(2)