Python tf.keras.utils.get_registered_object用法及代码示例

如果在 Keras 中注册，则返回与 name 关联的类。

用法

tf.keras.utils.get_registered_object(
    name, custom_objects=None, module_objects=None
)

参数

name 要查找的名称。
custom_objects 用于查找名称的自定义对象字典。通常，custom_objects 由用户提供。
module_objects 用于查找名称的自定义对象字典。通常，module_objects 由中级库实现者提供。

与'name' 关联的可实例化类，如果不存在此类，则为 None。

该函数是 Keras 序列化和反序列化框架的一部分。它将字符串映射到与其关联的对象以进行序列化/反序列化。

例子：

def from_config(cls, config, custom_objects=None):
  if 'my_custom_object_name' in config:
    config['hidden_cls'] = tf.keras.utils.get_registered_object(
        config['my_custom_object_name'], custom_objects=custom_objects)

相关用法

注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.keras.utils.get_registered_object。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。