(self, x_i, x_j, edge_index_i, size_i)
| 64 | return out |
| 65 | |
| 66 | def apply_edge(self, x_i, x_j, edge_index_i, size_i): |
| 67 | alpha = self.att_i(x_i) + self.att_j(x_j) |
| 68 | alpha = tf.nn.leaky_relu(alpha) |
| 69 | alpha = mp_ops.scatter_softmax(alpha, edge_index_i, size_i) |
| 70 | |
| 71 | x_j = x_j * tf.reshape(alpha, [-1, 1]) |
| 72 | return tf.reshape(x_j, [-1, self.dim]) |
| 73 | |
| 74 | def apply_node(self, aggr_out, x): |
| 75 | if self.improved: |