BVh!dZddlZddlmZddlmZddlmZedgGdd ej Z d Z d Z y) z6Tensor-like objects that are composed from tf.Tensors.N)pywrap_tensorflow)nest) tf_exportz__internal__.CompositeTensor)v1cLeZdZdZej dZdZdZdZ dZ y)CompositeTensora Abstract base class for Tensor-like objects that are composed from Tensors. Each `CompositeTensor` can be decomposed into a structured collection of component `tf.Tensor`s, and reconstructed from those components. The `tensorflow.python.util.nest` module has support for treating composite tensors as structure, which makes it easy to flatten and reconstruct composite tensors (or larger structures that contain composite tensors). E.g.: ```python ct = ... # Create a composite tensor. flat_list_of_tensors = nest.flatten(ct, expand_composites=True) transformed_list_of_tensors = ... # do something with the flat tensors. result = nest.pack_sequence_as(ct, transformed_list_of_tensors, expand_composites=True) ``` cDtt|jd)z/A `TypeSpec` describing the type of this value.z ._type_spec()NotImplementedErrortype__name__selfs \/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/framework/composite_tensor.py _type_speczCompositeTensor._type_spec-s! d!4!4 5]C DDcDtt|jd)aReturns a TypeSpec given a shape invariant (used by `tf.while_loop`). Args: shape: A `tf.TensorShape` object. The shape invariant for this `CompositeTensor`, or `None` if a default shape invariant should be used (based on the value of this `CompositeTensor`). Returns: A nested structure whose values are `tf.TensorShape` objects, specifying the shape invariants for the tensors that comprise this `CompositeTensor`. z._shape_invariant_to_type_specr )rshapes r_shape_invariant_to_type_specz-CompositeTensor._shape_invariant_to_type_spec2s)  :   => @@rc tjtj|dDcgc]}t|dd|j!c}}t t |Scc}w)zReturns a list of `Operation`s that consume this `CompositeTensor`. Returns: A list of `Operation`s. Raises: RuntimeError: If this method is called while executing eagerly. Texpand_compositesgraphN)rflattengetattr consumerslistset)r componentrs r _consumerszCompositeTensor._consumersDsa ddC  9gt , 8 I I  s$A(c8|jj|SN)r__tf_tracing_type__)rcontexts rr#z#CompositeTensor.__tf_tracing_type__Ts ?? . .w 77rc|S)aLConverts ResourceVariable components to Tensors. Override this method to explicitly convert ResourceVariables embedded in the CompositeTensor to Tensors. By default, it returns the CompositeTensor unchanged. Returns: A CompositeTensor with all its ResourceVariable components converted to Tensors. rs r_convert_variables_to_tensorsz-CompositeTensor._convert_variables_to_tensorsWs  KrN) r __module__ __qualname____doc__abcabstractpropertyrrr r#r'r&rrrrs8&EE@$ 8 rr) metaclassct|tr$t|jj |St j |s|St jt|dS)aRecursively replaces CompositeTensors with their components. Args: structure: A `nest`-compatible structure, possibly containing composite tensors. Returns: A copy of `structure`, where each composite tensor has been replaced by its components. The result will contain no composite tensors. Note that `nest.flatten(replace_composites_with_components(structure))` returns the same value as `nest.flatten(structure)`. Fr) isinstancer"replace_composites_with_componentsr_to_componentsr is_nested map_structure) structures rr0r0es] ?+ -++I6 88 >>) $    *I PPrc"|jSr")r')composite_tensors rconvert_variables_to_tensorsr7|s  7 7 99r) r*r+tensorflow.pythonrtensorflow.python.utilr tensorflow.python.util.tf_exportrABCMetarr0r7r&rrr<sL= /'6 )b1I I2IXP.:r