|View source on GitHub|
Registers an object with the Keras serialization framework.
tf.keras.utils.register_keras_serializable( package='Custom', name=None )
This decorator injects the decorated class or function into the Keras custom object dictionary, so that it can be serialized and deserialized without needing an entry in the user-provided custom object dict. It also injects a function that Keras will call to get the object's serializable string key.
Note that to be serialized and deserialized, classes must implement the
get_config() method. Functions do not have this requirement.
The object will be registered under the key 'package>name' where
defaults to the object name if not passed.
package: The package that this class belongs to.
name: The name to serialize this class under in this package. If None, the class's name will be used.
A decorator that registers the decorated class with the passed names.