fix: apply scale parameter inside exp in MultiScaleKernel

aidos-lab · Aug 22, 2024 · fa2017c · fa2017c
1 parent aba1474
commit fa2017c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torch_topological/nn/multi_scale_kernel.py b/torch_topological/nn/multi_scale_kernel.py
@@ -131,8 +131,8 @@ def forward(self, X, Y, p=2.):
             # diagram 2
             denom = self._dist(D1, self._mirror(D2), p)
 
-            M = torch.exp(-nom) / (8 * self.sigma)
-            M -= torch.exp(-denom) / (8 * self.sigma)
+            M = torch.exp(-nom / (8 * self.sigma))
+            M -= torch.exp(-denom / (8 * self.sigma))
 
             # sum over all points
             k_sigma += M.sum() / (8. * self.sigma * torch.pi)