Tensorflow使用tf.variable_scope(name)无法定义范围名称

时间:2018-11-22 20:04:41

标签: python tensorflow

我想使用tf.variable_scope(name)在此范围内重用变量,但似乎无法正常工作

with tf.variable_scope("D1"):
    ....

第一次调用此变量,该变量在作用域名称“ D1”下命名,但是第二次调用此方法,其重用= True,此作用域名称更改为D1_1 我不知道如何发生,如果您知道,请帮助我。 完整的代码块是:

with tf.variable_scope(name):
        if reuse:
            tf.get_variable_scope().reuse_variables()
        h0 = prelu(conv2d(image, self.c_dim, name='d' + branch + '_h0_conv', reuse=False),
                 name='d' + branch + '_h0_prelu', reuse=False)
        h1 = prelu(d_bn1(conv2d(h0, self.df_dim, name='d' + branch + '_h1_conv', reuse=False), reuse=reuse),
                 name='d' + branch + '_h1_prelu', reuse=False)
        h1 = tf.reshape(h1, [self.batch_size, -1])

您还可以在full code中找到完整的代码 在model.py脚本中

0 个答案:

没有答案