TensorFlow(一)创建图,启动图

2017-10-09  本文已影响18人  Oo晨晨oO

要使用TensorFlow, 我们选择的语言是python, 使用python的包管理工具Anaconda

配置安装

配置顺序:

  1. 安装Anaconda, 自带jupyter (或者使用pip来安装$pip3 install --upgrade pip $pip3 install jupyter)
  2. 安装TensorFlowpip3 install tensorflow

使用TensorFlow

这次仅仅实现创建图, 启动图

  1. 导入TensorFlow
  2. 创建常量op
  3. 创建一个矩阵乘法
  4. 定义一个会话
  5. 调用sess的run来执行矩阵乘法
import tensorflow as tf

#创建一个常量op
m1 = tf.constant([[3, 3]])
#创建一个常量op
m2 = tf.constant([[2], [3]])
#创建一个矩阵乘法
product = tf.matmul(m1, m2)
print(product)

#定义一个会话
sess = tf.Session()
#调用sess的run来执行矩阵乘法
result = sess.run(product)
print(result)
sess.close()

注意下半部分的代码可以使用with语法简化:

with tf.Session() as sess:
    result = sess.run(product)
    print(result)

关于with的语法, 可以参考这里
简而言之就是with可以不用再去管close了

上一篇 下一篇

猜你喜欢

热点阅读