651b7d63f09c73e5a093b64c97dd6266265ff930,torch_geometric/nn/functional/spline_utils.py,,weight_index,#Any#Any#Any#,113

Before Change

    // 1. Calculate offset.
    // TODO: more elegant, please!
    offset = [1]
    curr_kernel = 1
    for i in range(len(kernel_size) - 1):
        curr_kernel *= kernel_size[i]
        offset.append(curr_kernel)

After Change

    // kernel [k_1, k_2, k_3]` the offset is given by [k_2 * k_3, k_3, 1].

    // 1. Calculate offset.
    off = [reduce(lambda x, y: x * y, kernel_size[i:]) for i in range(1, dim)]
    off.append(1)
    off = torch.LongTensor([off]).t()

    // 2. Apply offset.

Instances