(self, inputs)
| 79 | self.layers = [layers.Dense(dim, activation=tf.nn.relu)] |
| 80 | |
| 81 | def aggregate(self, inputs): |
| 82 | embedding = inputs |
| 83 | for layer in self.layers: |
| 84 | embedding = layer(embedding) |
| 85 | return self.pool(embedding) |
| 86 | |
| 87 | def pool(self, inputs): |
| 88 | raise NotImplementedError() |