fix inverse_phat_square

lehner · Dec 10, 2023 · d44bfb4 · d44bfb4
1 parent ea0ed0e
commit d44bfb4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/gpt/algorithms/optimize/fourier_accelerate.py b/lib/gpt/algorithms/optimize/fourier_accelerate.py
@@ -44,7 +44,7 @@ def __init__(self, grid, base, dimensions=None):
             self.weight += c_mu_l
 
         # special consideration for zero
-        self.weight[0, 0, 0, 0] = (2.0 * np.pi) ** 2.0 / np.prod(
+        self.weight[coor[0:1]] = (2.0 * np.pi) ** 2.0 / np.prod(
             [grid.gdimensions[mu] for mu in dimensions]
         ) ** (2.0 / len(dimensions))