python - TensorFlow 2：在一次调用 TensorFlow 时获取多个张量值

Question

在 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 运行时的一次调用，就像 TensorFlow 1 代码一样？我试过检查自动流生成的代码，但这并不明显。

标签： pythontensorflowtensor