在TensorFlow 1中(不急于执行),我可以通过一次调用Session.run
来获得多个张量值,如下所示:
# The tensors variable is a list containing multiple tensors
tensor_values = sess.run(tensors)
这很有用,因为每次调用Session.run
(调用TensorFlow运行时)都会产生一些开销。
在TensorFlow 2中,我想获得相同的结果,同时使用tf.function
优化执行。因此,我可以执行以下操作:
@tf.function
def my_func(tensors):
return tensors
我的问题是:与TensorFlow 1代码相同,这是否导致对TensorFlow运行时的单个调用?我已经尝试检查自动流生成的代码,但是由此看来并不明显。