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