tensorflow: name_scope 和 variable_scope的差别

来源:互联网 发布:python rsa 加密 编辑:程序博客网 时间:2024/06/05 09:45

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

阅读全文
1 0
原创粉丝点击