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.