Here's how I went about the exercise from the last section.

Exercise 18: Factor out add_image_axes()

• Pull the code out of add_input_image() that adds the image Axes into a separate function. This can be re-used for adding the other images in the Figure.
• Save a copy of the Figure to verify that it still works.

Feel free to use my solution to the last exercise as a starting point: nn_viz_17.py

Ready? Go for it.