ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

tf.debugging.Assert

TensorFlow 2 version View source on GitHub

Asserts that the given condition is true.

If condition evaluates to false, print the list of tensors in data. summarize determines how many entries of the tensors to print.

# 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 ...

condition The condition to evaluate.
data The tensors to print out when condition is false.
summarize Print this many entries of each tensor.
name A name for this operation (optional).

assert_op An Operation that, when executed, raises a tf.errors.InvalidArgumentError if condition is not true.

Eager Compatibility

tf.errors.InvalidArgumentError if condition is not true