tensorflow: name_scope 和 variable_scope的差别

时间:2021-08-25 13:38:06

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里面也能共享variable