TensorFlow 2.0 Beta is available Learn more

tf.contrib.eager.nan_callback

View source on GitHub

A specialization of inf_nan_callback that checks for nans only.

tf.contrib.eager.nan_callback(
    op_type,
    inputs,
    attrs,
    outputs,
    op_name,
    action=tf.contrib.eager.ExecutionCallback.RAISE
)