TensorFlow 2.0 Beta is available Learn more

tf.contrib.eager.errstate

Context manager setting error state.

tf.contrib.eager.errstate(
    *args,
    **kwds
)

Example:

c = tf.math.log(0.)  # -inf

with errstate(inf_or_nan=ExecutionCallback.RAISE):
  tf.math.log(0.)  # <-- Raises InfOrNanError.

Args:

  • inf_or_nan: An ExecutionCallback determining the action for infinity (inf) and NaN (nan) values. A value of None leads to no change in the action of the condition.

Yields:

None.

Raises:

  • ValueError: If the value of any keyword arguments is invalid.