美文网首页
tf.Variable和tf.get_variable的区别

tf.Variable和tf.get_variable的区别

作者: 溪奇 | 来源:发表于2017-05-06 22:15 被阅读0次

    tf.Variable的参数列表为tf.Variable(name=None, initial_value, validate_shape=True, trainable=True, collections=None),返回一个由initial_value创建的变量
    tf.get_variable的参数列表为tf.get_variable(name, shape=None, initializer=None, dtype=tf.float32, trainable=True, collections=None),如果已存在参数定义相同的变量,就返回已存在的变量,否则创建由参数定义的新变量。
    所以tf.get_variable创建变量时,会进行变量检查,当设置为共享变量时(通过scope.reuse_variables()或tf.get_variable_scope().reuse_variables()),检查到第二个拥有相同名字的变量,就返回已创建的相同的变量;如果没有设置共享变量,则会报[ValueError: Variable varx alreadly exists, disallowed.]的错误。而tf.Variable()创建变量时,name属性值允许重复,检查到相同名字的变量时,由自动别名机制创建不同的变量。
    例如:
    with tf.name_scope('name_scope_1'):
    var1 = tf.get_variable(name='var1', shape=[1], initializer=None, dtype=tf.float32)
    var2 = tf.Variable(name='var2', initial_value=[1], dtype=tf.float32)
    var21 = tf.Variable(name='var2', initial_value=[2], dtype=tf.float32)
    输出name时,如下:
    var1:0
    name_scope_1/var2:0
    name_scope_1/var2_1:0

    with tf.name_scope('name_scope_2') as scope:
    scope.reuse_variables() #tf.get_variable_scope().reuse_variables()
    var1 = tf.get_variable(name='var1', shape=[1], initializer=None, dtype=tf.float32)
    var11 = tf.get_variable(name='var1')
    var2 = tf.Variable(name='var2', initial_value=[1], dtype=tf.float32)
    var21 = tf.Variable(name='var2', initial_value=[2], dtype=tf.float32)
    with tf.Session() as sess:
    print(var1.name)
    print(var11.name)
    print(var2.name)
    print(var21.name)
    输出name时,如下:
    var1:0
    var1:0
    name_scope_2/var2:0
    name_scope_2/var2_1:0

    相关文章

      网友评论

          本文标题:tf.Variable和tf.get_variable的区别

          本文链接:https://www.haomeiwen.com/subject/bttrtxtx.html