How to get the input shape of TensorFlow layers?:

[Ammar_Abdullah]

Continuing the discussion from How to get the input shape of TensorFlow layers?:

@Nada-Nada

Having a model instance as input, I would like to create a function that prints the input shape of the model layers. I am aware of model.layers() but this is not general enough for what I want. Layers in my models might not have a sequential order, and a layer might receive its input from a layer different than the one preceding it. I would also to have tensorops included in case they are present in the model.

[Kiran_Sai_Ramineni]

Hi @Ammar_Abdullah, You can iterate through the layers in the model and can get the input shapes of the layers. For example,

input1 = keras.Input(shape=(128,))
input2 = keras.Input(shape=(64,))
dense1 = keras.layers.Dense(64)(input1)
dense2 = keras.layers.Dense(32)(input2)
concat = keras.layers.Concatenate()([dense1, dense2])
output = keras.layers.Dense(10)(concat)

model = keras.Model(inputs=[input1, input2], outputs=output)

for layer in model.layers:
    print(layer.input)

Thank You.