tf.compat.v2.debugging.assert_all_finite

View source on GitHub

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

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

Same tensor as x.