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

Before Change

    def __call__(self, data):
        _, 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)

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