typo

7e5277b1 · rusty1s · c78c4cd8 · 7e5277b1
Commit 7e5277b1 authored Mar 25, 2018 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

torch_spline_conv/functions/degree.py torch_spline_conv/functions/degree.py +2 -2

No files found.
--- a/torch_spline_conv/functions/degree.py
+++ b/torch_spline_conv/functions/degree.py
 import torch
-def node_degree(edge_index, n, out=None):
+def node_degree(edge_index, num_nodes, out=None):
-    zero = torch.zeros(n, out=out)
+    zero = torch.zeros(num_nodes, out=out)
    one = torch.ones(edge_index.size(1), out=zero.new())
    return zero.scatter_add_(0, edge_index[0], one)