制作tf.编译失败时函数错误

Question

tensorflowstf.function使用Autograph将函数转换为图形表示，但不能转换所有代码。如果失败，它将抛出一个警告("tensorflow将按原样运行此代码…")。有没有办法使程序出错?这样，如果没有抛出错误，我可以确定我所有的函数都成功地转换为图形模式。

Answer 1

查看源代码，似乎有一个称为AUTOGRAPH_STRICT_CONVERSION的环境变量，您可以将其设置为大于零的某个整数值，以便在无法执行转换时强制签名引发错误。我无法在文档中找到对此的参考，因此它可能会在将来被删除或替换为API中更易于访问的功能，但要知道，您可以在代码开头使用这样的东西来启用它:

import os
os.environ['AUTOGRAPH_STRICT_CONVERSION'] = '1'
# TensorFlow code...