5e4bcd2876408cc4fabfaee595928306f90f555d,torch_geometric/transform/target_indegree.py,TargetIndegree,__call__,#TargetIndegree#Any#,7

Before Change

        _, col = data.index

        zero = torch.zeros(data.num_nodes, out=col.new().float())
        one = torch.ones(col.size(0), out=zero.new())

        degree = zero.scatter_add_(0, col, one)
        degree /= degree.max()
        degree = degree[col]

        if data.weight is None:
            data.weight = degree
        else:
            degree, weight = degree.unsqueeze(1), data.weight.unsqueeze(1)
            data.weight = torch.cat([degree, weight], dim=1)

        return data

After Change

        if pseudo is not None and self.cat:
            pseudo = pseudo.unsqueeze(-1) if pseudo.dim() == 1 else pseudo
            deg = deg.type_as(pseudo)
            data.weight = torch.cat([pseudo, deg], dim=-1)
        else:
            data.weight = deg

Instances