guides/understanding_masking_and_padding.py/CustomEmbedding/call
class CustomEmbedding:    def call(self, inputs):
        inputs = ops.cast(inputs, "int32")
        return ops.take(self.embeddings, inputs)