tensorflow只返回第一个参数

时间:2017-12-01 11:53:58

标签: python tensorflow

标题的确如此:

with tf.Session() as session:
   a = tf.placeholder(tf.float32)
   b = tf.placeholder(tf.float32)
   c = a+b
   print(session.run(c, feed_dict={a:3,b:2}))

上面的代码应该在运行时打印5。但它总是打印分配给a的任何内容。即使使用tf.add(a,b)而不是a+b,这也是相同的结果。

奇怪的是,上面的代码只在一台计算机上出现故障。它在其他任何地方都可以正常工作。

这是某种配置问题吗?谁能告诉我这里发生了什么?

背景资料:
我使用的是tensorflow的GPU版本 我使用conda安装了这个:conda install tensorflow-gpu
我在码头上运行这个。
如果我将占位符类型更改为int32

,这可以正常工作

更新
重新启动容器后,问题就消失了。但我仍然很想知道是什么导致了它。如何在不重新启动容器的情况下修复它?

0 个答案:

没有答案