I am trying to expand dimension:
import tensorflow as tf inp = tf.keras.layers.Input(shape=(1,)) inp = inp[..., tf.newaxis] decoder_input = inp output = tf.co