327ffc711693f1d0fe9ef6c40f086a33a9231019,chainer_chemistry/models/gat.py,GraphAttentionNetworks,update,#GraphAttentionNetworks#Any#Any#Any#,71

Before Change

        h_i = functions.broadcast_to(h_i, (mb, atom, atom,
                                           self.heads, self.hidden_dim))
        device_id = cuda.get_device_from_array(h_i.array).id
        h_j = functions.copy(h_i, device_id)
        h_j = functions.transpose(h_j, (0, 2, 1, 3, 4))

        // (minibatch, atom, atom, heads, out_dim * 2)
        e = functions.concat([h_i, h_j], axis=4)

After Change

                                           self.heads, self.hidden_dim))

        // (minibatch, atom, 1, heads, out_dim)
        h_j = functions.reshape(h, (mb, atom, 1, self.heads, self.hidden_dim))
        // (minibatch, atom, atom, heads, out_dim)
        h_j = functions.broadcast_to(h_j, (mb, atom, atom,
                                           self.heads, self.hidden_dim))

Instances