TensorFlow学习笔记--TensorFlow简介,常用基本操作

来源:互联网 发布:淘宝艾灸盒哪个牌子好 编辑:程序博客网 时间:2024/05/29 11:40

要将深度学习更快且更便捷地应用于新的问题中,选择一款深度学习工具是必不可少的步骤。 
TensorFlow是谷歌于2015年11月9日正式开源的计算框架。TensorFlow计算框架可以很好地支持深度学习的各种算法。

TensorFlow很好地兼容了学术研究和工业生产的不同需求。

  • 一方面,TensorFlow的灵活性使得研究人员能够利用它快速实现新的模型设计;
  • 另一方面,TensorFlow强大的分布式支持,对工业界在海量数据集上进行的模型训练也至关重要。作为谷歌开源的深度学习框架,TensorFlow包含了谷歌过去10年间对于人工智能的探索和成功的商业应用。

除了TensorFlow,目前还有一些主流的深度学习开源工具可以使用。每个工具都有着各自的特点,可以根据自己的需求和喜好选择适合自己的深度学习工具。比如我一开始学习深度学习的时候接触的是Caffe,然后TensorFlow开源之后了解到了TensorFlow的一些特性,感觉还是更喜欢TensorFlow的风格,当然也考虑到会使用一种以上的深度学习工具也算见不错的事情。

其它一些主流的深度学习工具有:

  • Caffe: http://caffe.berkeleyvision.org/(Convolutional Architecture for Fast Feature Embedding) BVLC
  • MXNet: http://mxnet.readthedocs.io/en/latest/(百度 DMLC(分布式机器学习社区) 简称”深盟”)
  • Torch: http://torch.ch/(Facebook Google DeepMind Twitter FAIR)
  • Theano: http://deeplearning.net/software/theano/(the LISA group at the University of Montreal(蒙特利尔))
  • TensorFlow: https://www.tensorflow.org/(Google)
  • CNTK(微软深度学习工具包 )
  • DeepLearning4J: http://deeplearning4j.org/
  • deepmat: https://github.com/kyunghyuncho/deepmat
  • Neon: http://neon.nervanasys.com/docs/latest/index.html
  • Eblearn: http://eblearn.sourceforge.net/
  • PyLearn: http://deeplearning.net/software/pylearn2/
  • chainer: https://github.com/pfnet/chainer

Bahrampour S, Ramakrishnan N, Schott L, et al. Comparative Study of Deep Learning Software Frameworks[J]. Computer Science, 2016.

这篇文章对现在流行的五个开源深度学习框架 caffe、Neon、TensorFlow、Theano、Torch 做了很严谨比较。作者开源了他们的比较 Benchmarks 代码:https://github.com/DL-Benchmarks/DL-Benchmarks。

文章比较了:可扩展性(extensibility)、hardware utilization(硬件利用率)和速度(speed)三个方面

  • 评估测试都是部署在单机上,对于多线程 CPU、GPU(Nvidia Titan X)都进行测试
  • 速度评估标准包括了梯度计算时间(gradient computation time)、前向传播时间(forward time)
  • 对于卷积神经网络,作者还对这几个深度框架支持的不同的卷积算法以及相应的性能表现做了实验

通过实验得出了以下结论

  • Theano、Torch 是最具扩展性的深度学习框架
  • 在 CPU 上的测试性能来看,Torch 最优,其次是 Theano
  • 在 GPU 上的性能表现,对于大规模卷积以及全连接网络,还是 Torch 最优,其次是 Neon
  • Theano 在部署和训练 LSTM 网络中夺得拔筹
  • caffe 是最容易测试评估性能的标准深度学习框架
  • 最后,TensorFlow 与 Theano 有些相似,是比较灵活的框架,但是其性能表现,目前还比不上上面的几个框架

但是,毕竟这篇文章已经是过去时了,那时候TensorFlow 还只能用 cuDNN v.2 版本,而我现在已经装的是v5.1版本,而且TensorFlow 也已经发布了1.0版本。现在各工具性能如何,还需要新的评测才能说明问题。


变量:创建、初始化、保存和加载

当训练模型时,用变量来存储和更新参数。变量包含张量 (Tensor)存放于内存的缓存区。建模时它们需要被明确地初始化,模型训练后它们必须被存储到磁盘。这些变量的值可在之后模型训练和分析是被加载。

本文档描述以下两个TensorFlow类。点击以下链接可查看完整的API文档: 
tf.Variable 类 
tf.train.Saver 类 
参考TensorFlow中文社区

- 添加神经层

输入参数有 inputsin_sizeout_size, 和 activation_function

# 添加层def add_layer(inputs, in_size, out_size, activation_function=None):            weights = tf.Variable(tf.random_normal([in_size, out_size]), name='weights')            biases = tf.Variable(tf.zeros([1, out_size]) + 0.1, name='biases')            y = tf.matmul(inputs, weights) + biases        if activation_function is None:            outputs = y        else:            outputs = activation_function(y)        return outputs
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10

- loss

  • loss = tf.reduce_mean(tf.reduce_sum(tf.square(ys - prediction), reduction_indices=[1]))
  • 分类问题的 loss 函数 cross_entropy 交叉熵 
    loss = tf.reduce_mean(-tf.reduce_sum(ys * tf.log(prediction), reduction_indices=[1]))

- 创建

当创建一个变量时,你将一个张量作为初始值传入构造函数Variable()。TensorFlow提供了一系列操作符来初始化张量,初始值是常量或是随机值。 
注意,所有这些操作符都需要你指定张量的shape。那个形状自动成为变量的shape。变量的shape通常是固定的,但TensorFlow提供了高级的机制来重新调整其行列数。

# Create two variables.weights = tf.Variable(tf.random_normal([784, 200], stddev=0.35), name="weights")biases = tf.Variable(tf.zeros([200]), name="biases")
  • 1
  • 2
  • 3
  • 1
  • 2
  • 3

- 初始化

变量的初始化必须在模型的其它操作运行之前先明确地完成。最简单的方法就是添加一个给所有变量初始化的操作,并在使用模型之前首先运行那个操作。 
使用tf.global_variables_initializer()添加一个操作对变量做初始化。记得在完全构建好模型并加载之后再运行那个操作。

# 7.初始化变量init = tf.global_variables_initializer()# tf.global_variables_initializer()是并行地初始化所有变量# 有时候会需要用另一个变量的初始化值给当前变量初始化,这就需要注意了# 用其它变量的值初始化一个新的变量时,使用其它变量的initialized_value()属性。# 你可以直接把已初始化的值作为新变量的初始值,或者把它当做tensor计算得到一个值赋予新变量。# w1 = tf.Variable(tf.random_normal([784, 200], stddev=0.35), name="w1")# w2 = tf.Variable(w1.initialized_value(), name="w2")# 8.启动图 (graph)sess = tf.Session()sess.run(init)
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12

- 由另一个变量初始化

你有时候会需要用另一个变量的初始化值给当前变量初始化。由于tf.global_variables_initializer()是并行地初始化所有变量,所以在有这种需求的情况下需要小心。用其它变量的值初始化一个新的变量时,使用其它变量的initialized_value()属性。你可以直接把已初始化的值作为新变量的初始值,或者把它当做tensor计算得到一个值赋予新变量。

w1 = tf.Variable(tf.random_normal([784, 200], stddev=0.35), name="w1")w2 = tf.Variable(w1.initialized_value(), name="w2")
0 0
原创粉丝点击