Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "e696cfd64ddbfe569eafe35c86e731f04271b0c0"
Commit fb2e852d authored by Jan Eric Lenssen's avatar Jan Eric Lenssen
Browse files

cleanuped, bp_to_adj now set by is_tensor() of values

parent 850bb170
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
spline_conv.py
View file @ fb2e852d
...@@ -16,8 +16,7 @@ def spline_conv( ...@@ -16,8 +16,7 @@ def spline_conv(
basis_kernel, basis_kernel,
basis_backward_kernel=None, basis_backward_kernel=None,
degree=1, degree=1,
bias=None, bias=None):
bp_to_adj=False):
if input.dim() == 1: if input.dim() == 1:
input = input.unsqueeze(1) input = input.unsqueeze(1)
...@@ -28,6 +27,7 @@ def spline_conv( ...@@ -28,6 +27,7 @@ def spline_conv(
# Get features for every end vertex with shape [|E| x M_in]. # Get features for every end vertex with shape [|E| x M_in].
output = input[col] output = input[col]
bp_to_adj = False if torch.is_tensor(values) else True
# Convert to [|E| x M_in] feature matrix and calculate [|E| x M_out]. # Convert to [|E| x M_in] feature matrix and calculate [|E| x M_out].
if output.is_cuda: if output.is_cuda:
......
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