51b53dcbab8ec7ab0b6e8a64284a919db2d2254a,torch_geometric/transform/local_cartesian.py,LocalCartesian,__call__,#LocalCartesian#Any#,8

Before Change

        max_n, _ = max_n.max(dim=1)

        max_e = max_n.gather(dim=0, index=row)
        weight *= 1 / (2 * max_e.unsqueeze(1))
        weight += 0.5

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

        return data

After Change

        cartesian /= 2 * tmp.max(dim=1, keepdim=True)[0][row]
        cartesian += 0.5

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

        return data

Instances