tensorflow tf.add是如何实现的？

Question

我希望更熟悉TensorFlow库的人可以帮助解决一个简单的问题。我想知道如何实现tensorflow添加操作。

其他tensorflow操作是注册和定义的内核，但是基本算术操作在哪里/如何处理？

https://github.com/tensorflow/tensorflow/tree/master/tensorflow/core/kernels

Answer 1

tf.add() Python函数是一个自动生成的包装函数（当前位于模块tensorflow.python.ops.gen_math_ops中），它将一个节点添加到当前默认的TensorFlow graph。

当您运行包含该节点的图形（通过tf.Session.run()）时，TensorFlow运行时将调用BinaryOp<Device, tensorflow::functor::add>的实例，该实例包含一些在所有分段二进制操作中通用的代码（例如，广播和参数验证），以及tensorflow::functor::add()的调用，它使用Eigen scalar_sum_op来执行添加。