View source on GitHub
|
Asserts that the given condition is true.
tf.debugging.Assert(
condition, data, summarize=None, name=None
)
If condition evaluates to false, print the list of tensors in data.
summarize determines how many entries of the tensors to print.
Args | |
|---|---|
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). |
Returns | |
|---|---|
assert_op
|
An Operation that, when executed, raises a
tf.errors.InvalidArgumentError if condition is not true.
|
Raises |
|---|
TF1 compatibility
When in TF V1 mode (that is, outside tf.function) Assert needs a control
dependency on the output to ensure the assertion executes:
# 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 compatibility
returns None