c04fe56d4c5d15554a9d9c7c2955851de6b37994,torch_geometric/nn/dense/diff_pool.py,DenseDiffPool,forward,#DenseDiffPool#Any#Any#,23

Before Change

        x = x.unsqueeze(0) if x.dim() == 2 else x
        adj = adj.unsqueeze(0) if adj.dim() == 2 else adj
        s = s.unsqueeze(0) if s.dim() == 2 else s
        batch_size = x.size(0)

        out = torch.matmul(s.transpose(1, 2), x)
        out_adj = torch.matmul(torch.matmul(s.transpose(1, 2), adj), s)

After Change

        reg = adj - torch.matmul(s, s.transpose(1, 2))
        reg = torch.norm(reg, p=2)
        reg = reg / reg.numel()

        return out, out_adj, reg

    def __repr__(self):

Instances