AVh#4dZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l m Z e d d g e je jd ddZe d g e jddZe dg dZy )z?Connects all half, float and double tensors to CheckNumericsOp.)context)dtypes)ops) array_ops)control_flow_ops) deprecation)dispatch) tf_exportzdebugging.assert_all_finiteverify_tensor_all_finite)v1Nc|tjd|d|}tjd|d|}t|||S)aAssert that the tensor does not contain any NaN's or Inf's. Args: t: Tensor to check. msg: Message to log on failure. name: A name for this operation (optional). x: Alias for t. message: Alias for msg. Returns: Same tensor as `t`. xtmessagemsg)rdeprecated_argument_lookupverify_tensor_all_finite_v2)rrnamerrs N/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/ops/numerics.pyr r sB ,,S!S!>> @tf.function ... def f(x): ... x = tf.debugging.assert_all_finite(x, 'Input x must be all finite') ... return x + 1 >>> f(tf.constant([np.inf, 1, 2])) Traceback (most recent call last): ... InvalidArgumentError: ... Args: x: Tensor to check. message: Message to log on failure. name: A name for this operation (optional). Returns: Same tensor as `x`. VerifyFiniter)rrN)r name_scopeconvert_to_tensor colocate_withrcheck_numericsrwith_dependencies)rrr verify_inputouts rrr2s. ~~dNQC0BD ac*A  1 B--aAl  . . ~q AcBB *BBB *s#-B/B5BB BBadd_check_numerics_opscTtjr tdg}tjj D]}|j D]}|jtjtjtjfvs?|j td|jdzt|j z}tj"|5t%j&||g}dddt)j*|S#1swYxYw)aConnect a `tf.debugging.check_numerics` to every floating point tensor. `check_numerics` operations themselves are added for each `half`, `float`, or `double` tensor in the current default graph. For all ops in the graph, the `check_numerics` op for all of its (`half`, `float`, or `double`) inputs is guaranteed to run before the `check_numerics` op on any of its outputs. Note: This API is not compatible with the use of `tf.cond` or `tf.while_loop`, and will raise a `ValueError` if you attempt to call it in such a graph. Returns: A `group` op depending on all `check_numerics` ops added. Raises: ValueError: If the graph contains any numeric operations in a control flow structure. RuntimeError: If called with eager execution enabled. @compatibility(eager) Not compatible with eager execution. To check for `Inf`s and `NaN`s under eager execution, call `tf.debugging.enable_check_numerics()` once before executing the checked operations. @end_compatibility zadd_check_numerics_ops() is not compatible with eager execution. To check for Inf's and NaN's under eager execution, call tf.debugging.enable_check_numerics() once before executing the checked operations.Nz`tf.add_check_numerics_ops() is not compatible with TensorFlow control flow operations such as `tf.cond()` or `tf.while_loop()`.:r)rexecuting_eagerly RuntimeErrorrget_default_graphget_operationsoutputsdtyperfloat16float32float64_get_control_flow_context ValueErrorrstr value_indexcontrol_dependenciesrrrgroup)check_opopoutputrs rr!r!Qs6     (  ! ! # 2 2 4 Ib** I &..&..&..I I ' ' ) 5?@ @''C-#f&8&8"99  % %h / I..vwGH( I I I I    ** I Is 'DD' )NNNNN)N)__doc__tensorflow.python.eagerrtensorflow.python.frameworkrrtensorflow.python.opsrrtensorflow.python.utilrr tensorflow.python.util.tf_exportr add_dispatch_supportdeprecated_endpointsr rr!rrr?s F+.++2.+6 ,.HIJ !!!"<=7>K7& (R0  1 : '()1+*1+r