Skip to content

Commit ae31f72

Browse files
committed
fix scatter_std bug
1 parent 3a58d6f commit ae31f72

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torch_scatter/composite/std.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ def scatter_std(src: torch.Tensor, index: torch.Tensor, dim: int = -1,
2626

2727
index = broadcast(index, src, dim)
2828
tmp = scatter_sum(src, index, dim, dim_size=dim_size)
29-
count = broadcast(count, tmp, dim).clamp_(1)
29+
count = broadcast(count, tmp, dim).clamp(1)
3030
mean = tmp.div(count)
3131

3232
var = (src - mean.gather(dim, index))

0 commit comments

Comments
 (0)