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