tf.keras.utils.CustomObjectScope  |  TensorFlow v2.0.0 (original) (raw)

tf.keras.utils.CustomObjectScope

Stay organized with collections Save and categorize content based on your preferences.

Provides a scope that changes to _GLOBAL_CUSTOM_OBJECTS cannot escape.

View aliases

Compat aliases for migration

SeeMigration guide for more details.

tf.compat.v1.keras.utils.CustomObjectScope

tf.keras.utils.CustomObjectScope(
    *args
)

Code within a with statement will be able to access custom objects by name. Changes to global custom objects persist within the enclosing with statement. At end of the with statement, global custom objects are reverted to state at beginning of the with statement.

Example:

Consider a custom object MyObject (e.g. a class):

    with CustomObjectScope({'MyObject':MyObject}):
        layer = Dense(..., kernel_regularizer='MyObject')
        # save, load, etc. will recognize custom object by name

Methods

__enter__

View source

__enter__()

__exit__

View source

__exit__(
    *args, **kwargs
)

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2020-10-01 UTC.