tf.compat.v1.trainable_variables
Возвращает все переменные, созданные с помощью trainable=True.
tf.compat.v1.trainable_variables(
scope=None
)
Перейти на TF2
Несовместимо с eager execution и tf.function. В частности, коллекции графа в TF2 объявлены устаревшими. Вместо этого создайте контейнер tf.Module для всего состояния модели, включая переменные. Затем можно получить список всех обучаемых переменных в tf.Module с помощью атрибута trainable_variables.
Описание
При передаче trainable=True конструктор Variable() автоматически добавляет новые переменные в коллекцию графа GraphKeys.TRAINABLE_VARIABLES. Эта вспомогательная функция возвращает содержимое этой коллекции.
| Аргументы | |
|---|---|
scope | (Необязательно.) Строка. Если указано, возвращаемый список фильтруется: в него включаются только элементы, атрибут name которых соответствует scope с использованием re.match. Если область видимости указана, элементы без атрибута name не возвращаются. Выбор re.match означает, что scope без специальных токенов фильтрует элементы по префиксу. |
| Возвращает | |
|---|---|
| Список объектов Variable. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/compat/v1/trainable_variables
