01b293b84fd2b1eeb80a01afe44d341d9c9c6486,torch_geometric/sparse/sparse.py,,SparseTensor,#Any#Any#Any#,5

Before Change

def SparseTensor(indices, values, size):
    if values.is_cuda:
        A = getattr(cuda_sparse, values.__class__.__name__)
    else:
        A = getattr(sparse, values.__class__.__name__)

    return A(indices, values, size)

After Change

        A = getattr(cuda_sparse, values.type().split(".")[-1])
    else:
        print(values)
        A = getattr(sparse, values.type().split(".")[-1])

    return A(indices, values, size)

Instances