tf.contrib.eager.inf_nan_callback

An execution callback that checks for infs and nans in output tensors.

tf.contrib.eager.inf_nan_callback(
    op_type, inputs, attrs, outputs, op_name, check_inf=True, check_nan=True,
    action=tf.contrib.eager.ExecutionCallback.RAISE
)

This callback can be used with tfe.add_execute_callback to check for invalid numeric values. E.g.,

tfe.add_execute_callback(tfe.inf_nan_callback)

Args
op_type	Name of the TFE operation type (e.g., MatMul).
inputs	The list of input tensors to the operation, currently unused by this callback.
attrs	Attributes of the TFE operation, as a tuple of alternating attribute names and attribute values.
outputs	The list of output tensors from the operation, checked by this callback for inf and nan values.
op_name	Name of the TFE operation. This name is set by client and can be None if it unset.
check_inf	(bool) Whether this callback should check for inf values in the output tensor values.
check_nan	(bool) Whether this callback should check for nan values in the output tensor values.
action	(ExecutionCallback) Action to be taken by the callback when inf or nan values are detected.

Raises
InfOrNanError	iff inf or nan values are seen in any of outputs and action is "raise".
ValueError	iff the value of action is invalid.