TensorFlow 2:在一次调用TensorFlow中获取多个张量值

时间:2019-12-18 11:35:45

标签: python tensorflow tensor

在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运行时的单个调用?我已经尝试检查自动流生成的代码,但是由此看来并不明显。

0 个答案:

没有答案