61d8eda8a6c2f4d54896cd242d3534509c6c1ac8,torch_geometric/utils/degree.py,,degree,#Any#Any#Any#,6

Before Change

    
    num_nodes = maybe_num_nodes(index, num_nodes)
    out = torch.zeros((num_nodes), dtype=dtype, device=index.device)
    return out.scatter_add_(0, index, out.new_ones((index.size(0))))

After Change

    :rtype: :class:`Tensor`
    
    num_nodes_int = maybe_num_nodes(index, num_nodes)
    out = torch.zeros((num_nodes_int), dtype=dtype, device=index.device)
    out_ones = torch.ones((index.size(0)), dtype=out.dtype, device=out.device)
    return out.scatter_add_(0, index, out_ones)

Instances