tf.keras.StatelessScope

Область видимости, предотвращающая любые обновления переменных Keras.

tf.keras.StatelessScope(
    state_mapping=None, collect_losses=False, initialize_variables=True
)

Значения переменных, используемые внутри области видимости, следует передавать через аргумент state_mapping — список кортежей (k, v), где k — это KerasVariable, а v — предполагаемое значение этой переменной (тензор бэкенда).

Обновлённые значения можно получить при выходе из области видимости с помощью value = scope.get_current_value(variable). В течение всего времени действия области видимости переменные не будут обновляться на месте.

Пример:

state_mapping = [(k, ops.ones(k.shape, k.dtype)) for k in model.weights]
with keras.StatelessScope(state_mapping) as scope:
    outputs = model.some_function(inputs)

# All model variables remain unchanged. Their new values can be
# collected via:
for k in model.weights:
    new_value = scope.get_current_value(k)
    print(f"New value for {k}: {new_value})

Методы

add_loss

Просмотреть исходный код

add_loss(
    loss
)

add_update

Просмотреть исходный код

add_update(
    update
)

get_current_value

Просмотреть исходный код

get_current_value(
    variable
)

__enter__

Просмотреть исходный код

__enter__()

__exit__

Просмотреть исходный код

__exit__(
    *args, **kwargs
)