text_input = tf.keras.Input(shape=(None,), name=\'text\') text_features = tf.keras.layers.Embedding(VOCAB_SIZE, 64, mask_zero=True)(text_input) text_features = tf.ke