Can you unroll a tensorflow model?

Question

I have a tf.keras.models.Model which takes in an image (256,256,1) and outputs an image (256,256,1). Let\'s call this model \'unet\'. I am interested