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
)