Skip to content

GitLab

Commit e58d83e7 authored by rusty1s's avatar rusty1s
Browse files

no print

parent 2a3dca8e
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
torch_scatter/std.py
View file @ e58d83e7
...@@ -6,7 +6,6 @@ from torch_scatter.utils.gen import gen ...@@ -6,7 +6,6 @@ from torch_scatter.utils.gen import gen
def scatter_std(src, index, dim=-1, out=None, dim_size=None, unbiased=True): def scatter_std(src, index, dim=-1, out=None, dim_size=None, unbiased=True):
src, out, index, dim = gen(src, index, dim, out, dim_size, fill_value=0) src, out, index, dim = gen(src, index, dim, out, dim_size, fill_value=0)
print('src', src.mean())
tmp = scatter_add(src, index, dim, None, dim_size) tmp = scatter_add(src, index, dim, None, dim_size)
count = scatter_add(torch.ones_like(src), index, dim, None, dim_size) count = scatter_add(torch.ones_like(src), index, dim, None, dim_size)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment