fix division by zero

702103ed · rusty1s · ff285368 · 702103ed
Commit 702103ed authored May 11, 2021 by rusty1s
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torch_scatter/composite/std.py torch_scatter/composite/std.py +1 -1

No files found.
--- a/torch_scatter/composite/std.py
+++ b/torch_scatter/composite/std.py
@@ -34,6 +34,6 @@ def scatter_std(src: torch.Tensor, index: torch.Tensor, dim: int = -1,

    if unbiased:
        count = count.sub(1).clamp_(1)
-    out = out.div(count).sqrt()
+    out = out.div(count + 1e-6).sqrt()

    return out