テンソルフローはどのように実装されていますか？

Question

私は、TensorFlowライブラリをよく知っている人が簡単な質問に役立つことを望んでいました。テンソルフローの追加操作の実装方法を知りたい。

他のテンソルフローは登録され、定義されたカーネルですが、基本的な算術演算はどのように扱われますか？

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

Answer 1

tf.add() Pythonの関数は、現在のデフォルトTensorFlow graphにノードを追加する（現在モジュールtensorflow.python.ops.gen_math_opsで）自動的に生成されたラッパー関数です。

あなたは（tf.Session.run()を経由して）、そのノードを含むグラフを実行すると、TensorFlowランタイムがあるBinaryOp<Device, tensorflow::functor::add>のインスタンスを、呼び出されますが、（放送と引数の検証のためなど）は、すべての要素ごとのバイナリ操作に共通であるいくつかのコードが含まれており、 tensorflow::functor::add()の呼び出しは、Eigenのscalar_sum_opを使用して加算を実行します。