How to get the input shape of TensorFlow layers?

Hi @Nada-Nada, I have tried in an alternative way by creating a dictionary in the class by adding the layer name and input and able to get those. please refer to this gist for working code example. Thank You.