Spec-Zone.ru › TensorFlow 1.15

tf.contrib.framework.model_variable

Получает существующую переменную модели с заданными параметрами или создаёт новую.

tf.contrib.framework.model_variable(
    name, shape=None, dtype=tf.dtypes.float32, initializer=None, regularizer=None,
    trainable=True, collections=None, caching_device=None, device=None,
    partitioner=None, custom_getter=None, use_resource=None,
    synchronization=tf.VariableSynchronization.AUTO,
    aggregation=tf.VariableAggregation.NONE
)
Аргументы
name имя новой или существующей переменной.
shape форма новой или существующей переменной.
dtype тип новой или существующей переменной (по умолчанию DT_FLOAT).
initializer инициализатор переменной, если она создаётся.
regularizer функция (Tensor -> Tensor или None); результат её применения к новой переменной будет добавлен в коллекцию GraphKeys.REGULARIZATION_LOSSES и может быть использован для регуляризации.
trainable Если True также добавить переменную в коллекцию графа GraphKeys.TRAINABLE_VARIABLES (см. tf.Variable).
collections Список имён коллекций, в которые будет добавлена переменная. Обратите внимание, что переменная всегда добавляется также в коллекции GraphKeys.GLOBAL_VARIABLES и GraphKeys.MODEL_VARIABLES.
caching_device Необязательная строка устройства или функция, описывающая, где переменная должна кэшироваться для чтения. По умолчанию — устройство переменной.
device Необязательное устройство для размещения переменной. Может быть строкой или функцией, вызываемой для получения устройства для переменной.
partitioner Необязательная функция, принимающая полностью определённую TensorShape и тип Variable создаваемой переменной и возвращающая список разделов для каждой оси (в настоящее время может быть разделена только одна ось).
custom_getter Функция, позволяющая переопределить внутренний метод get_variable и должна иметь тот же сигнатуру.
use_resource Если True использовать ResourceVariable вместо Variable.
synchronization Указывает, когда распределённая переменная будет агрегирована. Допустимые значения — константы, определённые в классе tf.VariableSynchronization. По умолчанию синхронизация установлена в AUTO, и текущий DistributionStrategy выбирает, когда синхронизироваться.
aggregation Указывает, как распределённая переменная будет агрегирована. Допустимые значения — константы, определённые в классе tf.VariableAggregation.
Возвращаемое значение
Созданная или существующая переменная.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/framework/model_variable

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API