阅读(11.9k) 书签 (0)

TensorFlow如何用常量替换变量

2018-10-17 17:46 更新

tf.graph_util.convert_variables_to_constants

convert_variables_to_constants (  
    sess ,  
    input_graph_def ,  
    output_node_names ,  
    variable_names_whitelist = None ,  
    variable_names_blacklist = None
  )

定义在:tensorflow/python/framework/graph_util_impl.py

用相同值的常量替换图中的所有变量.

如果您有一个包含变量(Variable)操作的训练有素的图表,那么可以将它们全部转换为具有相同值的 CONST 操作.这样可以用一个 GraphDef 文件完全描述网络,并允许删除许多与加载和保存变量相关的操作.

参数:

  • sess:包含变量的 Active TensorFlow 会话.
  • input_graph_def:具有网络的 GraphDef 对象.
  • output_node_names:图表结果节点的名称字符串列表.
  • variable_names_whitelist:要转换的变量名称集合(默认情况下,所有变量都将被转换).
  • variable_names_blacklist:要省略转换为常量的变量名的集合.

返回值:

GraphDef 包含原版的一个简化的版本.