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 никогда не возвращаются, если указан область. Выбор 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
