tensorflow中get_variable函数的使用

get_variable()

tf.get_variable(name, shape, initializer): name就是变量的名称,shape是变量的维度,initializer是变量初始化的方式,初始化的方式有以下几种:

tf.constant_initializer:常量初始化函数

tf.random_normal_initializer:正态分布

tf.truncated_normal_initializer:截取的正态分布

tf.random_uniform_initializer:均匀分布

tf.zeros_initializer:全部是0

tf.ones_initializer:全是1

tf.uniform_unit_scaling_initializer:满足均匀分布,但不影响输出数量级的随机值

举例:

import tensorflow as tf

a1 = tf.get_variable(name='a1', shape=[3,3,3,16], dtype=tf.float32,initializer=tf.truncated_normal_initializer(stddev=0.1, dtype=tf.float32))

with tf.Session() as sess:
    sess.run(tf.initialize_all_variables())
    print(sess.run(a1))

运行结果为:
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/qq_29566629/article/details/90053803