小编car*_*rop的帖子

我试图保存一个张量数组，该数组必须使用装饰器计算到函数中@tf.function，这使得函数内的所有张量都变成张量图，因此成为不可迭代的对象。例如，在下面的最小代码中，我想知道是否可以使用函数内的代码将张量保存到文件中foo()。

@tf.function
def foo(x):
    # code for saving x


a=tf.constant([1,2,3])
foo(a)

我想知道在 TensorFlow 2.4 中迭代 tf.data.Dataset 的最有效方法是什么。

我正在使用典型的：

for example in dataset:
    code

但是，我测量了挂机时间，并且由于我的数据集很大，因此计算循环需要太多时间。还有其他选择可以减少计算时间吗？