小编AHG*_*AHG的帖子

如何在 TensorFlow 图中正确引发异常

我想tf.errors.InvalidArgumentError根据图形模式(在 TensorFlow 服务中)中的输入张量的值引发异常。

目前我正在使用tf.debugging.assert_all_finite并且效果很好。由于我不是对错误检查做出断言,而是根据输入引发异常,因此最好引发显式异常。

我的问题归结为:

  • 如何有条件地执行不返回张量的代码
  • 如何引发 tf.errors 异常。

这样做的正确方法是什么?

编辑:更多细节。我想在不使用 tf.debugging 的情况下重新创建以下逻辑(除非这实际上是正确的方法)。

目前我正在检查是否存在NaN这样的值:

assert_op = tf.debugging.assert_all_finite(
    input_data,
    'Cant have nans at beginning or end'
)
Run Code Online (Sandbox Code Playgroud)

python python-3.x tensorflow tensorflow-serving tensorflow2.0

5
推荐指数
1
解决办法
3121
查看次数