(edge_index, size)
| 31 | |
| 32 | @staticmethod |
| 33 | def norm(edge_index, size): |
| 34 | edge_weight = tf.ones([tf.shape(edge_index)[1], 1]) |
| 35 | |
| 36 | def deg_inv_sqrt(i): |
| 37 | deg = mp_ops.scatter_add(edge_weight, edge_index[i], size[i]) |
| 38 | return deg ** -0.5 |
| 39 | |
| 40 | return tuple(map(deg_inv_sqrt, [0, 1])) |
| 41 | |
| 42 | def __call__(self, x, edge_index, size=None, **kwargs): |
| 43 | norm = self.norm(edge_index, size) |