tensorflow: name_scope 和 variable_scope的差别

Variable sharing 简介

因为我目前对variable_scope的理解,这个功能主要是是针对Variable sharing 来做的,所以先介绍一下variable sharing:
主要有两种方法实现

  • 直接在各个ops,function之间传递variable reference.
  • 把variable 封装在variable_scope/name_scope 中

variable_scop/name_scope简介

这两种scope最直接的影响是: 所有在scope下面创建的variable都会把这个scope的名字作为前缀。比如如下代码

with tf.name_scope("my_scope"):
    v1 = tf.get_variable("var1", [1], dtype=tf.float32)
    v2 = tf.Variable(1, name="var2", dtype=tf.float32)

with tf.variable_scope("my_scope"):
    v3 = tf.get_variable("var1", [1], dtype=tf.float32)
    v4 = tf.Variable(1, name="var2", dtype=tf.float32)
print("v1:",v1)
print("v2:",v2)
print("v3:",v3)
print("v4:",v4)

输出结果如下

v1: <tf.Variable 'var1:0' shape=(1,) dtype=float32_ref>
v2: <tf.Variable 'my_scope/var2:0' shape=() dtype=float32_ref>
v3: <tf.Variable 'my_scope/var1:0' shape=(1,) dtype=float32_ref>
v4: <tf.Variable 'my_scope_1/var2:0' shape=() dtype=float32_ref>

name_scope 与 variable_scope的差别

在上面的输出中,v1没有my_scope的前缀。这个就是name_scope 和variable_scope的差别:tf.get_variable 会忽略 name_scope. name_scope 会对operation 起作用。

所以这就允许我们使用name_scope给operation分类。同时使用variable_scope来区分variable.

转载自:https://blog.csdn.net/scotthuang1989/article/details/77477026

猜你喜欢

转载自blog.csdn.net/Tong_jy/article/details/81174878
今日推荐