Was this section helpful?
tf.keras.Model
for full control over model architecture and forward pass.tf.function
, which automatically converts Python code into a TensorFlow graph for performance, relevant for understanding how subclassed models are optimized.