babe2e9ace7bf78a46ff2011d95d97a08461cf2a,chainer_chemistry/models/gat.py,GAT,__call__,#GAT#Any#Any#,167

Before Change

        if isinstance(adj, Variable):
            w_adj = adj.data
        else:
            w_adj = adj
        is_real_node = self.xp.sum(w_adj, axis=(1, 2)) > 0
        w_adj = Variable(w_adj, requires_grad=False)

        g_list = []
        for step in range(self.n_layers):
            h = self.update(h, w_adj, step)
            if self.concat_hidden:
                g = self.readout(h, h0, step, is_real_node)
                g_list.append(g)

After Change

        if self.concat_hidden:
            return functions.concat(g_list, axis=1)
        else:
            g = self.readout_layers[](h, h0)
            return g

Instances