tf.compat.v1.trainable_variables
Возвращает все переменные, созданные с помощью trainable=True.
tf.compat.v1.trainable_variables(
scope=None
)
Переход на TF2
Несовместим с выполнением eager и tf.function. В частности, коллекции графов устарели в TF2. Вместо этого, пожалуйста, создайте контейнер tf.Module для всего состояния вашей модели, включая переменные. Затем вы можете перечислить все обучаемые переменные в вашем tf.Module через атрибут trainable_variables.
Описание
При передаче trainable=True, конструктор Variable() автоматически добавляет новые переменные в коллекцию графа GraphKeys.TRAINABLE_VARIABLES. Эта удобная функция возвращает содержимое этой коллекции.
| Аргументы | |
|---|---|
scope | (Необязательно.) Строка. Если указана, результирующий список фильтруется, чтобы включать только элементы, чьи атрибут name соответствует scope с использованием re.match. Элементы без атрибута name никогда не возвращаются, если указан scope. Выбор 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/versions/r2.9/api_docs/python/tf/compat/v1/trainable_variables
