from tensorflow import keras

input = keras.Input((2,))