python - 循环数据集时 tf.function 中的异常执行顺序

Question

给定以下代码片段。

import tensorflow as tf

a=tf.Variable(0)

@tf.function
def f():
    dataset = tf.data.Dataset.from_tensor_slices([1, 2, 3])
    for i in dataset:
        a.assign(i)
        tf.print(a)
    tf.print(a)

f()

如果在 TF 2.0 中执行，结果为1 2 3 3.

如果在 TF 2.1 中执行，结果为1 2 3 0.

为什么两个版本之间存在差异以及为什么第二个输出有效？

标签： pythontensorflowtensorflow2.0

Answer 1

这已在 TF 2.2 中修复。