Parameters ---------- inputs_shape : tuple the shape of inputs tensor
(self, inputs_shape)
| 375 | return s.format(classname=self.__class__.__name__, **self.__dict__) |
| 376 | |
| 377 | def build(self, inputs_shape): |
| 378 | """ |
| 379 | Parameters |
| 380 | ---------- |
| 381 | inputs_shape : tuple |
| 382 | the shape of inputs tensor |
| 383 | """ |
| 384 | |
| 385 | self.embeddings = self._get_weights( |
| 386 | "embeddings", |
| 387 | shape=(self.vocabulary_size, self.embedding_size), |
| 388 | init=self.E_init, |
| 389 | ) |
| 390 | |
| 391 | # @tf.function |
| 392 | def forward(self, inputs): |