tf.contrib.model_pruning.strip_pruning_vars_fn
Удаляет переменную маски из графа.
tf.contrib.model_pruning.strip_pruning_vars_fn(
input_graph_def, output_node_names
)
Заменяет тензор masked_weight на поэлементное умножение маски и соответствующей переменной веса.
| Аргументы | |
|---|---|
input_graph_def | GraphDef, в котором переменные были преобразованы в константы. Обычно это результат tf.graph_util.convert_variables_to_constant() |
output_node_names | Список имен строк для узлов результата графа |
| Возвращает | |
|---|---|
| GraphDef, из которого были удалены переменные, связанные с обрезкой |
© 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/model_pruning/strip_pruning_vars_fn
