Subclass definitions of TensorFlow models with customizable hidden layers

Question

I am learning about Models subclass definitions in TensorFlow

A pretty straightforward definition will be something like this

class MyNetwork1(tf.keras