首页 > 其他 > 详细

Tensorflow细节-P42张量的概念及使用

时间:2019-10-01 17:26:22      阅读:86      评论:0      收藏:0      [点我收藏+]

1、运行以下代码

import tensorflow as tf
a = tf.constant([1.0, 2.0], name="a")
b = tf.constant([2.0, 3.0], name="b")
result = a + b
print result

sess = tf.InteractiveSession ()
print(result.eval())
sess.close()

得到
技术分享图片
其中,add与代码中的add有关,0表示第一个输出,图中的shape(2, )则表示一个长度为2的一维数组

2、讲一个特别重要的概念——result.get_shape()可以获得张量的维度信息

3、见到result.eval(session=sess)不要奇怪,sess = tf.Interactivesession()也不要奇怪,其他的就算了

4、至于

g=tf.Graph()
with g.device('/gpu:0'):
  result = a+b

这种概念就不讲了

Tensorflow细节-P42张量的概念及使用

原文:https://www.cnblogs.com/liuboblog/p/11615641.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!