Commit fa633471 authored by rusty1s's avatar rusty1s
Browse files

no inline

parent a2c9a29b
...@@ -72,10 +72,10 @@ class SplineConv(object): ...@@ -72,10 +72,10 @@ class SplineConv(object):
# Weight root node separately (if wished). # Weight root node separately (if wished).
if root_weight is not None: if root_weight is not None:
out += torch.mm(src, root_weight) out = out + torch.mm(src, root_weight)
# Add bias (if wished). # Add bias (if wished).
if bias is not None: if bias is not None:
out += bias out = out + bias
return out return out
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