Spec-Zone.ru › TensorFlow 1.15

tf.contrib.framework.filter_variables

Фильтрует список переменных с использованием регулярных выражений.

tf.contrib.framework.filter_variables(
    var_list, include_patterns=None, exclude_patterns=None, reg_search=True
)

Сначала включает переменные в соответствии со списком include_patterns. Затем исключает переменные в соответствии со списком exclude_patterns.

Например, можно получить список переменных с весами всех свёрточных слоёв (в зависимости от определения сети) следующим образом:

variables = tf.contrib.framework.get_model_variables()
conv_weight_variables = tf.contrib.framework.filter_variables(
    variables,
    include_patterns=['Conv'],
    exclude_patterns=['biases', 'Logits'])
Аргументы
var_list список переменных.
include_patterns список регулярных выражений для включения. По умолчанию None, что означает, что все переменные выбираются в соответствии с правилами включения. Переменная включается, если она соответствует любому из include_patterns.
exclude_patterns список регулярных выражений для исключения. По умолчанию None, что означает, что все переменные выбираются в соответствии с правилами исключения. Переменная исключается, если она соответствует любому из exclude_patterns.
reg_search логическое значение. Если True (по умолчанию), выполняется re.search для поиска совпадений (т.е. шаблон может совпадать с любой подстрокой имени переменной). Если False, выполняется re.match (т.е. регулярное выражение должно совпадать с началом имени переменной).
Возвращает
отфильтрованный список переменных.

© 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/filter_variables

Spec-Zone.ru

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