(self, dim, improved=False, aggr='add', **kwargs)
| 41 | class GATConv(conv.Conv): |
| 42 | |
| 43 | def __init__(self, dim, improved=False, aggr='add', **kwargs): |
| 44 | super(GATConv, self).__init__(aggr=aggr, **kwargs) |
| 45 | |
| 46 | self.dim = dim |
| 47 | self.improved = improved |
| 48 | |
| 49 | self.fc = tf.layers.Dense(dim, use_bias=False) |
| 50 | self.att_i = Attention(dim) |
| 51 | self.att_j = Attention(dim) |
| 52 | |
| 53 | def __call__(self, x, edge_index, size=None, **kwargs): |
| 54 | if isinstance(x, tf.Tensor): |