Hi, I am working with residual networks. I want to view features of intermediated layer as an image. How do I do that if
               def residual_unit:
                return  res1, outAfterMask, mask

when I call that in my model:
                 self.res1 = resudal_unit(128, 128)

How do I save mask and view as an image. What I think if I can save them as an numpy array in .hf5 file and then view it using matplotlib. I am not sure how to save using a numpy array. Is it when I call the method in my model ? Any suggestions will be helpful