TensorFlow变量初始化

1、tf.Variable()

输入:

a = tf.Variable(initial_value = 1, trainable = True)

输出:

<tf.Variable 'Variable:0' shape=() dtype=int32_ref>

定义变量后,可以再赋值:

a = tf.ones((2,2))

2、tf.get_variable()

tf.get_variable 可以实现变量重复使用。

import tensorflow as tf

with tf.variable_scope("scope1"):
    w1 = tf.get_variable("w1", shape=[])
    w2 = tf.Variable(0.0, name="w2")
with tf.variable_scope("scope1", reuse=True):
    w1_p = tf.get_variable("w1", shape=[])
    w2_p = tf.Variable(1.0, name="w2")

print(w1 is w1_p, w2 is w2_p)
#输出
#True  False

3、tf.variable.scope()

同2。


参考:

  1. tf.Variable_csdn
  2. tensorflow学习笔记(二十三):variable与get_variable;
已标记关键词 清除标记
相关推荐
©️2020 CSDN 皮肤主题: 酷酷鲨 设计师:CSDN官方博客 返回首页