Having a model instance as input, I would like to create a function that prints the input shape of the