tf.debugging.assert_all_finite

Assert that the tensor does not contain any NaN's or Inf's.

@tf.function def f(x):  x = tf.debugging.assert_all_finite(x, 'Input x must be all finite')  return x + 1
f(tf.constant([np.inf, 1, 2])) Traceback (most recent call last):  InvalidArgumentError: ...

x Tensor to check.
message Message to log on failure.
name A name for this operation (optional).

Same tensor as x.