tf.reduce_sum结合placeholder作为输入形状时GPU运行失败
TensorFlow GPU环境下
tf.reduce_sum结合Placeholder形状输入的全零输出问题 我之前碰到过TensorFlow在GPU运行时的一个棘手问题:当使用tf.reduce_sum处理以placeholder定义形状的大张量时,输出结果会变成全零,而且这个问题只在GPU环境下触发,CPU环境下完全正常。
经过一番排查和版本验证,我整理了这些关键信息:
- 该问题已在TensorFlow 1.14.0中修复(更早版本是否修复尚未验证)
- TensorFlow 1.7.0中这个问题仍然存在
- 我编写了Colab笔记本,可在谷歌GPU硬件上复现此bug(可通过构造大张量+placeholder定义形状+调用
tf.reduce_sum的场景复现) - 最初我错误怀疑问题出在
tf.gather,后来准确定位到是tf.reduce_sum结合placeholder作为输入形状的特定场景导致的
内容的提问来源于stack exchange,提问作者sdnr
相关产品推荐
相关产品推荐

