Updating param_quad.

AlgorithmicDataReduction · Aug 10, 2023 · 2af6bbe · 2af6bbe
1 parent 664e54b
commit 2af6bbe
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torch_quadconv/utils/quadrature.py b/torch_quadconv/utils/quadrature.py
@@ -94,11 +94,11 @@ def param_quad(input_points, num_points, ratio=[], base=[], const=False):
     for i in range(spatial_dim):
         #linearly spaced nodes
         if base[i] == None:
-            nodes.append(torch.linspace(coord_min[i], coord_max[i], num_points*ratio[i]))
+            nodes.append(torch.linspace(coord_min[i], coord_max[i], round(num_points*ratio[i])))
 
         #logarithmically spaced nodes
         else:
-            p = torch.logspace(0, 1, steps=num_points*ratio[i], base=base[i])
+            p = torch.logspace(0, 1, steps=round(num_points*ratio[i]), base=base[i])
             p = coord_min[i] + ((coord_max[i]-coord_min[i])/(base[i]-1))*(p-1)
 
             nodes.append(p)