通过 CPU/GPU 的占位符获取不同的值

2023-12-25

当我运行下面的代码时,我认为我得到了错误的结果。
OVER GPU

import tensorflow as tf
sess = tf.Session()
a = tf.placeholder(tf.float32)
b = tf.placeholder(tf.float32)
adder_node = a + b
print(sess.run(adder_node, {a: 3, b: 4.5}))
print(sess.run(adder_node, {a: [1, 3], b: [2, 4]}))

Output:
3.0
[1.3.]

我认为真实的结果应该是;
7.5
[ 3. 7.]

OVER CPU

import tensorflow as tf
sess = tf.Session()
with tf.device('/cpu:0'):
    a = tf.placeholder(tf.float32)
    b = tf.placeholder(tf.float32)
    adder_node = a + b
print(sess.run(adder_node, {a: 3, b: 4.5}))
print(sess.run(adder_node, {a: [1, 3], b: [2, 4]}))

Output:
7.5
[1.3.]

张量流版本:
'1.3.0'
Python版本:
Python 2.7.12

当我在 CPU 和 GPU 上运行时,得到不同的结果。 预先感谢,任何帮助将不胜感激......


我编译了你的代码并得到了真实的结果。 我也在使用 张量流版本: ‘1.3.0’ Python版本: Python 2.7.12

代码及结果截图 https://i.stack.imgur.com/Vf04i.png

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

通过 CPU/GPU 的占位符获取不同的值 的相关文章

随机推荐