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 包含原版的一个简化的版本.

以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号

pFad - Phonifier reborn

Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.

Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.


Alternative Proxies:

Alternative Proxy

pFad Proxy

pFad v3 Proxy

pFad v4 Proxy