conv.py

# import torch


def spline_conv(x,
                edge_index,
                pseudo,
                weight,
                kernel_size,
                is_open_spline,
                degree=1,
                root_weight=None,
                bias=None):
    pseudo = pseudo.unsqueeze(-1) if pseudo.dim() == 1 else pseudo