如何在 tensorflow 的源代码中打印（跟踪）

Question

I have a small trial to print out something within tensorflow's kernel.

ex, I would like to print something within

https://www.tensorflow.org/api_docs/python/tf/train/GradientDescentOptimizer

Then I search for the corresponding sources in my laptop where Ubuntu 16.04 has been installed.

Below are what I found.

I think gradient_descent.py within either

/usr/local/lib/python3.5/dist-packages/tensorflow/contrib/optimizer_v2

or

/usr/local/lib/python3.5/dist-packages/tensorflow/python/training

are used. Hence I modify them both.

However, I cannot see the modifications.

I wonder why it failed.

Is there any way the trace Tensorflow source code (I know C & C++ part needs to be re-compile so focus on Python now)?

Thanks in advance.

Answer 1

Check CurrentStackTrace API provided by TensorFlow util . An example use can be found here .