Objective: build a neural net with skip connections, where the number of blocks is a parameter, and we subclass tf.keras.Model.
tf.keras.Model