TensorFlow中维护的集合列表
在一个计算图中,可以通过集合(collection
)来管理不同类别的资源。比如通过 tf.add_to_collection
函数可以将资源加入一个 或多个集合中,然后通过 tf.get_collection
获取一个集合里面的所有资源(如张量,变量,或者运行TensorFlow程序所需的队列资源等等)
集合名称 |
集合内容 |
使用场景 |
tf.GraphKeys.VARIABLES |
所有变量 |
持久化TensorFlow模型 |
tf.GraphKeys.TRAINABLE_VARIABLES |
可学习的变量(一般指神经网络中的参数) |
模型训练、生成模型可视化内容 |
tf.GraphKeys.SUMMARIES |
日志生成相关的张量 |
TensorFlow计算可视化 |
tf.GraphKeys.QUEUE_RUNNERS |
处理输入的QueueRunner |
输入处理 |
tf.GraphKeys.MOVING_AVERAGE_VARIABLES |
所有计算了滑动平均值的变量 |
计算变量的滑动平均值 |
- TensorFlow中的所有变量都会被自动加入
tf.GraphKeys.VARIABLES
集合中,通过 tf.global_variables()
函数可以拿到当前计算图上的所有变量。拿到计算图上的所有变量有助于持久化整个计算图的运行状态。
- 当构建机器学习模型时,比如神经网络,可以通过变量声明函数中的
trainable
参数来区分需要优化的参数(比如神经网络的参数)和其他参数(比如迭代的轮数,即超参数),若trainable = True
,则此变量会被加入tf.GraphKeys.TRAINABLE_VARIABLES
集合。然后通过 tf.trainable_variables
函数便可得到所有需要优化的参数。TensorFlow中提供的优化算法会将tf.GraphKeys.TRAINABLE_VARIABLES
集合中的变量作为 默认的优化对象。
- 变量的类型是不可以改变的。
- 变量的维度一般是不能改变的,除非设置参数
validate_shape = False
(很少去改变它)
TensorFlow中的tf.Variable
函数随机数和常数的生成:
函数名 |
随机数分布 |
主要参数 |
tf.random_normal |
正态分布 |
平均值、标准差、取值类型 |
tf.truncated_normal |
满足正态分布的随机值,但若随机值偏离平均值超过2个标准差,则这个数会被重新随机 |
平均值、标准差、取值类型 |
tf.random_uniform |
平均分布 |
最大、最小值、取值类型 |
tf.random_gamma |
Gramma分布 |
形状参数alpha、尺度参数beta、取值类型 |
函数名 |
功能 |
示例 |
tf.zeros |
产生全0的数组 |
tf.zeros([2, 3],tf.int32) |
tf.ones |
产生全1的数组 |
tf.ones([2, 3],tf.int32) |
tf.fill |
产生一个全部为给定数组的数组 |
tf.fill([2,3], 9) |
tf.constant |
产生一个给定值的常量 |
tf.constant([2,3,4]) |
TensorFlow 中的tf.get_variable
变量初始化函数
初始化函数 |
功能 |
主要参数 |
tf.constant_initializer |
将变量初始化为给定常数 |
常数的取值 |
tf.random_normal_initializer |
将变量初始化为满足正态分布的随机值 |
正态分布的均值和标准差 |
tf.truncated_normal_initializer |
将变量初始化为满足正态分布的随机值,但若随机值偏离平均值超过2个标准差,则这个数会被重新随机 |
正态分布的均值和标准差 |
tf.random_uniform_initializer |
将变量初始化为满足平均分布的随机值 |
最大、最小值 |
tf.uniform_unit_scaling_initializer |
将变量初始化为满足平均分布但不影响输出数量级的随机值 |
factor(产生随机值时乘以的系数) |
tf.zeros_initializer |
将变量初始化为全0 |
变量维度 |
tf.ones_initializer |
将变量初始化为全1 |
变量维度 |
-
tf.clip_by_value
函数将张量限定在一定的范围内:
sess = tf.InteractiveSession()
v = tf.constant([[1., 2., 3.], [4., 5., 6.]])
tf.clip_by_value(v, 2.5, 4.5).eval()
array([[ 2.5, 2.5, 3. ],
[ 4. , 4.5, 4.5]], dtype=float32)
-
tf.log
对张量所有元素进行对数运算
array([[ 0. , 0.69314718, 1.09861231],
[ 1.38629436, 1.60943794, 1.79175949]], dtype=float32)
-
tf.greater
的输入是两个张量,此函数会比较这两个张量中的每一个元素,并返回比较结果;
当输入维度不一致时会进行广播(broadcasting)
v1 = tf.constant([1., 2., 3., 4.])
v2 = tf.constant([4., 3., 2., 1.])
f = tf.greater(v1, v2)
f.eval()
Out[11]:
array([False, False, True, True], dtype=bool)
-
tf.where
函数有三个参数:
第一个选择条件根据,当选择条件为True
时,会选择第二个参数中的值,否则使用第三个参数中的值:
tf.where(f, v1, v2).eval()
array([ 4., 3., 3., 4.], dtype=float32)