f175fefd0a9ad8d55fb889e5d9e40fd942837bd2,torch_geometric/nn/conv/gat_conv.py,GATConv,__init__,#GATConv#Any#Any#Any#Any#Any#Any#Any#,49

Before Change

        self.__alpha__ = None

        self.weight = Parameter(torch.Tensor(in_channels,
                                             heads * out_channels))
        self.att = Parameter(torch.Tensor(1, heads, 2 * out_channels))

        if bias and concat:
            self.bias = Parameter(torch.Tensor(heads * out_channels))

After Change

        self.lin = Linear(in_channels, heads * out_channels, bias=False)

        self.att_i = Parameter(torch.Tensor(1, heads, out_channels))
        self.att_j = Parameter(torch.Tensor(1, heads, out_channels))

        if bias and concat:
            self.bias = Parameter(torch.Tensor(heads * out_channels))
        elif bias and not concat:

Instances