Exemple #1
0
 def call(self, inputs, mask=None, **kwargs):
     output = K.identity(inputs[0])
     if self.return_masked:
         return [
             output,
             K.cast(self.compute_mask(inputs, mask)[0], K.floatx())
         ]
     return output
Exemple #2
0
 def call(self, inputs):
     return [
         super(TokenEmbedding, self).call(inputs),
         K.identity(self.embeddings)
     ]