guides/making_new_layers_and_models_via_subclassing.py/Linear/get_config
class Linear:    def get_config(self):
        config = super().get_config()
        config.update({"units": self.units})
        return config