(self, dim, *args, **kwargs)
| 75 | class BasePoolAggregator(BaseAggregator): |
| 76 | |
| 77 | def __init__(self, dim, *args, **kwargs): |
| 78 | super(BasePoolAggregator, self).__init__(dim, *args, **kwargs) |
| 79 | self.layers = [layers.Dense(dim, activation=tf.nn.relu)] |
| 80 | |
| 81 | def aggregate(self, inputs): |
| 82 | embedding = inputs |