f4622e59a087d04bc0628b52c33f0cc9f296223e,torch_geometric/nn/conv/film_conv.py,FiLMConv,message,#FiLMConv#Any#Any#Any#,83

Before Change

        return out

    def message(self, x_j, beta, gamma) -> Tensor:
        if self.act is None:
            return self.lin(x_j)
        else:
            return self.act( gamma * self.lin(x_j) + beta )

    def __repr__(self):
        return "{}({}, dim={})".format(self.__class__.__name__, self.channels,
                                       self.dim)

After Change

    def message(self, x_j: Tensor, beta_i: Tensor, gamma_i: Tensor) -> Tensor:
        out = gamma_i * x_j + beta_i
        if self.act is not None:
            out = self.act(out)
        return out

    def __repr__(self):
        return "{}({}, {}, num_relations={})".format(self.__class__.__name__,

Instances