We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3a58d6f commit ae31f72Copy full SHA for ae31f72
torch_scatter/composite/std.py
@@ -26,7 +26,7 @@ def scatter_std(src: torch.Tensor, index: torch.Tensor, dim: int = -1,
26
27
index = broadcast(index, src, dim)
28
tmp = scatter_sum(src, index, dim, dim_size=dim_size)
29
- count = broadcast(count, tmp, dim).clamp_(1)
+ count = broadcast(count, tmp, dim).clamp(1)
30
mean = tmp.div(count)
31
32
var = (src - mean.gather(dim, index))
0 commit comments