为什么TensorFlow中的tf.reduce_sum方法会在图中创建另一个节点?该节点如何使用?

时间:2019-07-16 22:20:09

标签: python tensorflow graph reduce

当我使用tf.reduce_sum方法时,TensorFlow将一个节点“ sum”添加到图中。此外,根据我提供或不提供的参数,该图还会获得一个“常数”节点,该节点带有张量的等级和在方法内部传递的尺寸;如果包含参数轴,该节点将包含“ reduction_indices”例如,如果轴= 0,则将减小尺寸;

之后,TensorFlow如何使用这些额外的节点(“常量”或“缩减索引”)?为什么它创建一个常数或reduce_indices,而不仅创建一个具有不同值的常数张量?

此外,我阅读了math_ops.py文件,并在调用reduce_sum函数时调试了几次python代码,并多次输入了不同的参数以查看所创建的额外节点。

这是轴参数是否存在的代码。

{{1}}

以下是一些相关链接,但它们没有讨论图形的内部构造:

1。https://www.tensorflow.org/api_docs/python/tf/math/reduce_sum

2。How does reduce_sum() work in tensorflow?

0 个答案:

没有答案