目录:

前言

在练习tensorflow的时候发现了很多很有意思的基本问题,写个帖子记录一下,既方便了回顾,又方便了同学学习/

正文

tf.Session()和tf.InteractiveSession()的区别

官方tutorial是这么说的:

The only difference with a regular Session is that an InteractiveSession installs itself as the default session on
construction. The methods Tensor.eval() and Operation.run() will use that session to run ops.

翻译一下就是:tf.InteractiveSession()是一种交替式的会话方式,它让自己成为了默认的会话,也就是说用户在单一会话的情境下,不需要指明用哪个会话也不需要更改会话运行的情况下,就可以运行起来,这就是默认的好处。这样的话就是run和eval()函数可以不指明session啦。

对比一下:

import tensorflow as tf
import numpy as np

importtensorflow as tf
import numpy as np

testa=tf.constant([[1., 2., 3.],[4., 5., 6.]])
testb=np.float32(np.random.randn(3,2))
testc=tf.matmul(testa,testb)
init=tf.global_variables_initializer()
sess=tf.Session()
print (testc.eval())

上面的代码编译是错误的,显示错误如下:

ValueError: Cannot evaluate tensor using `eval()`: No default session is registered. Use `with sess.as_default()` or pass an explicit session to `eval(session=sess)`
import tensorflow as tf
import numpy as np

testa=tf.constant([[1., 2., 3.],[4., 5., 6.]])
testb=np.float32(np.random.randn(3,2))
testc=tf.matmul(testa,testb)
init=tf.global_variables_initializer()
sess=tf.InteractiveSession()
print (testc.eval())

而用InteractiveSession()就不会出错,简单来说InteractiveSession()等价于:

sess=tf.Session()
with sess.as_default():

换句话说,如果说想让sess=tf.Session()起到作用,一种方法是上面的

sess=tf.Session()
with sess as default:
    print (testc.eval())

另外一种方法是

sess=tf.Session()
print (c.eval(session=sess))

其实还有一种方法也是with,如下:
复制代码

import tensorflow as tf
import numpy as np

testa=tf.constant([[1., 2., 3.],[4., 5., 6.]])
testb=np.float32(np.random.randn(3,2))
testc=tf.matmul(a,b)
init=tf.global_variables_initializer()
with tf.Session() as sess:
    #print (sess.run(c))
    print(c.eval())

总结

tf.InteractiveSession()默认自己就是用户要操作的会话(session),而tf.Session()没有这个默认,因此用eval()启动计算时需要指明使用那个会话(session)

GitHub 加速计划 / te / tensorflow
184.55 K
74.12 K
下载
一个面向所有人的开源机器学习框架
最近提交(Master分支:2 个月前 )
a49e66f2 PiperOrigin-RevId: 663726708 2 个月前
91dac11a This test overrides disabled_backends, dropping the default value in the process. PiperOrigin-RevId: 663711155 2 个月前
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐