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