基于TensorFlow的深度学习系列教程 2——常量Constant

Posted 2021-02-02 xing901022

前面介绍过了Tensorflow的基本概念，比如如何使用tensorboard查看计算图。本篇则着重介绍和整理下Constant相关的内容。

基于TensorFlow的深度学习系列教程 1——Hello World!

常量的概念

在tensorflow中，数据分为几种类型: 常量Constant、变量Variable、占位符Placeholder。其中：

• 常量：用于存储一些不变的数值，在计算图创建的时候，调用初始化方法时，直接保存在计算图中
• 变量：模型训练的参数，比如全连接里面的W和bias
• 占位符：就是模型每次训练时的样本，当计算图固定时，只需要替换占位符里面的内容，就可以重新计算了。

概念上跟spark的DAG图差不多，不过图的模式更固定一些，不像spark还分为action和transform。

常量的简单使用

下面这个例子就是常量最简单的使用例子了，定义a和b两个常量，输出x。x=a+b。

import tensorflow as tf

a = tf.constant(2)
b = tf.constant(3)
x = tf.add(a, b)

with tf.Session() as sess:
    writer = tf.summary.FileWriter('./graphs', sess.graph)
    print(sess.run(x))

writer.close()

得到的计算图如下：

选中constant可以发现，它的值直接写在定义里面了。

常量的初始化

1 固定初始化、0或1初始化

最常用的初始化方法，就是直接在声明的时候赋予一个初始值，也可以根于指定的shape进行0和1的填充

import tensorflow as tf

# tf.constant(value, dtype=None, shape=None, name='Const', verify_shape=False)

# 常量的创建
# [2 2]
a = tf.constant([2, 2], name='vector')
# [[0 1] [2 3]]
b = tf.constant([[0, 1], [2, 3]], name='b')

# 也可以直接初始化成0或者1
# [[0 0 0] [0 0 0]]
zero1 = tf.zeros([2, 3], tf.int32)
# [[0 0] [0 0]]
zero2 = tf.zeros_like([[0, 1], [2, 3]])
# [[1 1 1] [1 1 1]]
one1 = tf.ones([2, 3], tf.int32)
# [[1 1] [1 1]]
one2 = tf.ones_like([[0, 1], [2, 3]])

# 基于填充创建
# [[8 8 8] [8 8 8]]
fill1 = tf.fill([2, 3], 8)

# 基于序列创建
# [10. 11. 12. 13.]
lnspace1 = tf.linspace(10., 13., 4, name='linspace')
# [ 3  7 11 15]
range1 = tf.range(3, 18, 4)
# [ 3  8 13]
range2 = tf.range(3, 18, 5)
# [ 0  1  2  3  4  5  6  7  8  9 10 11 12 13 14 15 16 17]
range3 = tf.range(18)

with tf.Session() as sess:
    print(sess.run(a))
    print(sess.run(b))

    print(sess.run(zero1))
    print(sess.run(zero2))
    print(sess.run(one1))
    print(sess.run(one2))

    print(sess.run(fill1))

    print(sess.run(lnspace1))

    print(sess.run(range1))
    print(sess.run(range2))
    print(sess.run(range3))

tensorflow在设计时，尽量模仿numpy，因此很多函数都很类似。不过有一些操作tf中还是无法支持的，比如map:

import tensorflow as tf
import numpy as np
"""
0.0
3.3333333333333335
6.666666666666667
10.0
"""
for a in np.linspace(0., 10., 4):
    print(a)
"""
TypeError: Tensor objects are not iterable when eager execution is not enabled. To iterate over this tensor use tf.map_fn.
"""
for a in tf.linspace(0., 10., 4):
    print(a)

常量的随机初始化

另一种常用的初始化方法就是指定随机方法进行初始化。

import tensorflow as tf
import cv2

# 初始化服从指定正态分布的数值
# [ 2.3021064   0.4199094  -0.03323628  0.47499242  0.36770386 -0.7848035 -0.70948434 -0.35462353  0.75125676  0.50364155]
r1 = tf.random_normal([10], mean=0.0, stddev=1.0, dtype=tf.float32)

# 产生截断的正态分布，如果与均值差值超过两倍，就重新生成
# [ 1.785729    0.5161861   0.3950558   1.5795906   0.25945508 -1.5349426 -0.00732355  0.14366971 -0.7726713  -0.2694001 ]
r2 = tf.truncated_normal([10])

# 产生low和high之间的均匀分布
# [-0.54088783 -2.957581    1.8622065  -2.7436473   0.8000214   2.087247  2.5148878  -0.19671392  0.9098282   1.6573    ]
r3 = tf.random_uniform([10], minval=-3, maxval=3, dtype=tf.float32)

# 随机打乱
# [4 2 1 5 3]
r4 = tf.random_shuffle([1, 2, 3, 4, 5])

# 随机裁剪,一般用在图像上
# [-1.6676509  -2.3372912  -0.39069057  2.044036   -2.0961857 ]
r5 = tf.random_crop(r3, [5])

# 图片例子
img = cv2.imread('tensorboard.jpg')
cv2.imshow('origin', img)

# 多项式
multinomial1 = tf.multinomial([[0.99], [0.2]], 10)
multinomial2 = tf.multinomial([[0, 0.02, 0.99], [0, 0.99, 0.2]], 10)
# r7 = tf.random_gamma([])

with tf.Session() as sess:
    print(sess.run(r1))
    print(sess.run(r2))
    print(sess.run(r3))
    print(sess.run(r4))
    print(sess.run(r5))
    print(sess.run(multinomial1))
    print(sess.run(multinomial2))

    img_tf = tf.convert_to_tensor(img)
    distorted_image = tf.random_crop(img_tf, [300, 300, 3])
    img_np = distorted_image.eval()
    cv2.imshow('random', img_np)
    cv2.waitKey(0)

里面的random_gamma没见过应用的场景，所以也没有细致的研究。