(self, aggr_out, origin, t)
| 70 | return norm_i * norm_j * x_j |
| 71 | |
| 72 | def apply_node(self, aggr_out, origin, t): |
| 73 | v = self.vs[0 if self.shared_weights else t] |
| 74 | out = v(origin) |
| 75 | out = aggr_out + out |
| 76 | if self.act: |
| 77 | out = self.act(out) |
| 78 | return out |