e60669c5aa467d5c11a508c01b37c4ed8e352fa6,torch_geometric/nn/conv/supergat_conv.py,SuperGATConv,__init__,#SuperGATConv#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#,107

Before Change

                                             heads * out_channels))

        if self.attention_type == "MX":
            self.att = Parameter(torch.Tensor(1, heads, 2 * out_channels))
        elif self.attention_type == "SD":
            self.register_parameter("att", None)
        else:
            raise ValueError

After Change

            self.att_r = Parameter(torch.Tensor(1, heads, out_channels))
        else:  // self.attention_type == "SD"
            self.register_parameter("att_l", None)
            self.register_parameter("att_r", None)

        self.att_x = self.att_y = None  // x/y for self-supervision

        if bias and concat:

Instances