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
def call(self, inputs): return [ super(TokenEmbedding, self).call(inputs), K.identity(self.embeddings) ]