Python tf.debugging.Assert用法及代码示例

断言给定条件为真。

用法

tf.debugging.Assert(
    condition, data, summarize=None, name=None
)

如果 condition 评估为 false，则在 data 中打印张量列表。 summarize 确定要打印的张量条目数。

注意：应该使用这个函数的输出。如果不是，则会记录警告或引发错误。要将输出标记为已使用，请调用其 .mark_used() 方法。

TF1 兼容性

当处于 TF V1 模式时(即在 tf.function 之外)，断言需要对输出的控制依赖以确保断言执行：

# Ensure maximum element of x is smaller or equal to 1
assert_op = tf.Assert(tf.less_equal(tf.reduce_max(x), 1.), [x])
with tf.control_dependencies([assert_op]):
  ... code using x ...

eager模式兼容性

返回None。