Tensorflow: Math变量初始化

# Tensorflow: Math变量初始化
import tensorflow as tf

# Method 1:
# 1. 使用xxxx()方法生成一个Tensor
# 2. 将这个Tensor作为tf.Variable()中的initial_value参数的入参, 来初始化这个变量
A = tf.Variable(tf.truncated_normal([2, 2], mean=0.0, stddev=1.0, dtype=tf.float32, seed=10))

# Method 2:
# 1. 使用xxxx_initializer()创建一个初始化方法,但并不生成具体的Tensor
# 2. 使用在tf.get_variable()和tf.variable_scope()等方法中, 其中的initializer参数指定为初始化方法
#    则在环境中创建或获取的变量都将使用这个方法进行初始化

# Method 2-tf.get_variable():
initializer_op = tf.truncated_normal_initializer(mean=0.0, stddev=1.0, seed=10, dtype=tf.float32)
B = tf.get_variable(name='B', shape=[2, 2], initializer=initializer_op)

# Method 2-with tf.variable_scope()-tf.get_variable():
with tf.variable_scope("test", initializer=initializer_op):
    C = tf.get_variable("C", shape=[2, 2])

sess = tf.InteractiveSession()
sess.run(tf.global_variables_initializer())
print(A.eval())
print(B.eval())
print(C.eval())
sess.close()

猜你喜欢

转载自blog.csdn.net/xky1306102chenhong/article/details/81131209