I want to integrate simple python code with in the tensorflow graph.
I'm not sure if it's feasible. If feasible, please suggest how to integrate it.
Use case is, I want feed the output of intermediate tensor as an input to my python code and then python code output to another tensor.
Does @tf.function can be use here?