tf.debugging.assert_all_finite

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.