tf.variable_creator_scope

Область, определяющая функцию создания переменной, используемую функцией variable().

@tf_contextlib.contextmanager
tf.variable_creator_scope(
    variable_creator
)

Ожидается, что variable_creator будет функцией со следующим сигнатуром:

def variable_creator(next_creator, **kwargs)

Создатель, предположительно, в конечном итоге вызовет next_creator для создания переменной, если он хочет создать переменную, а не вызывать Variable или ResourceVariable напрямую. Это помогает сделать создателей композируемыми. Создатель может выбрать создание нескольких переменных, вернуть уже существующие переменные или просто зарегистрировать, что переменная была создана, и делегировать дальнейшее создание следующим создателям в очереди. Создатели также могут изменять ключевые аргументы, видимые следующими создателями.

Пользовательские методы получения в области переменных в конечном итоге обратятся к этим пользовательским создателям, когда они создают переменные.

Допустимые ключевые аргументы в kwds:

• initial_value: Значение по умолчанию, Tensor, или Python-объект, преобразуемый в Tensor, которое является начальным значением для переменной. Начальное значение должно иметь указанную форму, если validate_shape не установлено в False. Также может быть вызываемой функцией без аргументов, которая возвращает начальное значение при вызове. В этом случае dtype должно быть указано. (Обратите внимание, что функции инициализатора из init_ops.py должны быть сначала привязаны к форме, прежде чем будут использованы здесь.)
• trainable: Если True, по умолчанию, GradientTape автоматически отслеживает использование этой переменной.
• validate_shape: Если False, позволяет переменной быть инициализированной значением неизвестной формы. Если True, по умолчанию, форма initial_value должна быть известна.
• caching_device: Необязательная строка устройства, описывающая, где переменная должна быть кэширована для чтения. По умолчанию — устройство переменной. Если не None, кэшируется на другом устройстве. Типичное использование — кэширование на устройстве, где находятся операции, использующие переменную, для устранения дублирования копирования через Switch и другие условные операторы.
• name: Необязательное имя переменной. По умолчанию — 'Variable' и автоматически получает уникальное имя. dtype: Если установлено, initial_value будет преобразован в указанный тип. Если None, либо тип данных будет сохранён (если initial_value является тензором), либо convert_to_tensor определит.
• constraint: Функция ограничения, применяемая к переменной после обновлений некоторыми алгоритмами.
• synchronization: Указывает, когда распределенная переменная будет агрегирована. Принятые значения — константы, определённые в классе tf.VariableSynchronization. По умолчанию синхронизация установлена в AUTO и текущий DistributionStrategy выбирает, когда синхронизировать.
• aggregation: Указывает, как распределённая переменная будет агрегирована. Принятые значения — константы, определённые в классе tf.VariableAggregation.

Этот набор может расширяться со временем, поэтому важно, чтобы подпись создателей была такой, как указано выше.

Возвращает:

Область, в которой активен создатель