TensorFlow中给op重新设定name属性以便在恢复模型时读取

在TensorFlow官方文档中有这么一段代码

b = tf.Variable(tf.zeros([1]))
W = tf.Variable(tf.random_uniform([1, 2], -1.0, 1.0))
y = tf.matmul(W, x_data) + b 
loss = tf.reduce_mean(tf.square(y - y_data))

如果是变量b或者变量w给它添加name属性可以这样:

b = tf.Variable(tf.zeros([1]), name='bias1')

如果是y想要给它添加name属性就可以用:

tf.identity(y, name='y')

恢复模型时就可以根据name来找到这个变量了

    y = graph.get_tensor_by_name("y:0")
发布了22 篇原创文章 · 获赞 15 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/qq_28566521/article/details/88861203
今日推荐