(self, x, edge_index, size=None, **kwargs)
| 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) |
| 45 | filter_out = x[0] |
| 46 | for k in range(self.K): |
| 47 | gather_x, gather_norm, = \ |
| 48 | self.gather_feature([[filter_out, x[1]], norm], edge_index) |
| 49 | filter_out = self.apply_edge(gather_x[1], |
| 50 | gather_norm[0], |
| 51 | gather_norm[1]) |
| 52 | filter_out = mp_ops.scatter_(self.aggr, |
| 53 | filter_out, |
| 54 | edge_index[0], |
| 55 | size=size[0]) |
| 56 | filter_out = self.apply_node(filter_out) |
| 57 | out = self.fc(filter_out) |
| 58 | return out |
| 59 | |
| 60 | def apply_edge(self, x_j, norm_i, norm_j): |
| 61 | return norm_i * norm_j * x_j |
nothing calls this directly
no test coverage detected