tf.contrib.eager.errstate

View source on GitHub

Context manager setting error state.

@contextlib.contextmanager
tf.contrib.eager.errstate(
    inf_or_nan=None
)

Example:

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

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

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.

ValueError If the value of any keyword arguments is invalid.