TensorFlow笔记

2 篇文章 0 订阅
1 篇文章 0 订阅

参考:
TENSORFLOW从入门到精通之——TENSORFLOW基本操作
http://www.tensorflownews.com/2018/03/28/tensorflow_base/
Tensorflow是一种计算图模型,即用图的形式来表示运算过程的一种模型。Tensorflow程序一般分为图的构建和图的执行两个阶段。

  • 图的构建,每次运算的的结果以及原始的输入数据都可称为一个节点(operation ,缩写为op)
import tensorflow as tf
m1=tf.constant([3,5])#定义的节点(operation,缩写为op)
m2=tf.constant([2,4])
result=tf.add(m1,m2)#通过节点构建的图/矩阵相加
print(result)#其并没有输出矩阵相加的结果,而是输出了一个包含三个属性的Tensor

图的构建

  • 图的执行,也就是在会话(session)中执行图模型中定义好的运算。
sess=tf.Session()#启动默认图模型
 print(sess.sun(result)) #运行图模型,传入参数result,执行矩阵加法,打印相加的结果[5,9]
 sess.close()#关闭会话session

另外一种写法:

with tf.session() as sess:
	res = sess.run([result])
print(res)

如下程序2-4所示,Tensorflow中使用with…device语句来指定GPU或CPU资源执行操作:

with tf.session() as sess:
	with tf.device(''/gpu:2"):#指定第二个GPU资源来运行下面的op(节点),指定第N个 "/gpu:N"
		m1 = tf.constant([3,5])
		m2 = tf.constant([2,4])
		result = tf.add(m1,m2)
		res = sess.run([result])
print(res)
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值