将numpy数组分配给从GraphDef导入的变量

时间:2018-06-20 19:38:42

标签: tensorflow

我正在对精简图形进行一些图形编辑,而没有为变量分配操作。问题是我似乎无法弄清楚如何在导入的图形中获取变量的tf.Variable实例。

graph.get_operation_by_name('my_var') # returns tf.Operation
graph.get_tensor_by_name('my_var:0')  # returns tf.Tensor

这些都不能用作tf.assign()的操作数。换句话说,我试图找出如何分配给使用VariableV2加载到图形中的tf.import_graph_def()节点。

我可以想到的一种怪异方法是在导入GraphDef之前,将带有分配节点的NodeDef原型手动添加到GraphDef。这样,导入后,我的图将包含必要的分配节点。

如果当前图中不存在分配节点,是否有更简单的方法将会话图中的变量设置为给定的正确形状的numpy数组?

0 个答案:

没有答案