BVh\zdZddlmZddlmZddlmZddlmZddlmZddl m Z dZ d Z d Z d Zd Zd ZdZy)z/Python dataset sparse tensor utility functions.)nest)dtypes) sparse_tensor)tensor) tensor_shape) sparse_opscLtdtj|DS)zChecks for sparse tensor. Args: classes: a structure of objects that identify the dataset item classes Returns: `True` if `classes` contains a sparse tensor type and `False` otherwise. c3@K|]}|tjuyw)N)r SparseTensor).0cs R/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/data/util/sparse.py zany_sparse..!s LQ-,, , Ls)anyrflatten)classess r any_sparsers Ldll76K L LLc tj|ttj|tj|Dcgc]-\}}|tj urt jn|/c}}}|Scc}}w)aConverts sparse tensor shapes to their physical shapes. Args: shapes: a structure of shapes to convert. classes: a structure of objects that identify the dataset item classes Returns: a structure matching the nested structure of `shapes`, containing `tensor_shape.unknown_shape()` at positions where `classes` contains `tf.sparse.SparseTensor` and matching contents of `shapes` otherwise )rpack_sequence_asziprrr r unknown_shape)shapesrshaper rets ras_dense_shapesr$sp f$,,v. W0EF' %'(=+E+E&El  "5P' # * 's2B c tj|ttj|tj|Dcgc])\}}|tj urt jn|+c}}}|Scc}}w)aConverts sparse tensor types to `dtypes.variant`. Args: types: a structure of types to convert. classes: a structure of objects that identify the dataset item classes Returns: a structure matching the nested structure of `types`, containing `dtypes.variant` at positions where `classes` contains `tf.sparse.SparseTensor` and matching contents of `types` otherwise )rrrrrr rvariant)typesrtyr rs ras_dense_typesr!7sk et||E*DLL,AB& "a]777fnnR?& # * &s.A> ctj|ttj|tj|tj|tj|Dcgc]=\}}}}|tj ur"t j|||jn|?c}}}}}|Scc}}}}w)aDeserializes sparse tensors. Args: tensors: a structure of tensors to deserialize. types: a structure that holds information about types of `tensors` shapes: a structure that holds information about shapes of `tensors` classes: a structure of objects that identify the dataset item classes Returns: `tensors` with any serialized sparse tensors replaced by their deserialized version. )dtyperank) rrrrrr rdeserialize_sparsendims) tensorsrrrrr rr rs rdeserialize_sparse_tensorsr(Js e%( ,,w e!4dll66J ,,w %!&& !62ua m(( (##F"5;;G.45& # *&s0AB>c tj|tj|Dcgc]<}t|tj rtj nt j>c}Scc}w)a/Gets classes for a structure of tensors. Args: tensors: the tensor structure to get classes for. Returns: a structure matching the nested structure of `tensors`, containing `tf.sparse.SparseTensor` at positions where `tensors` contains a sparse tensor and `tf.Tensor` otherwise. )rrr isinstancerr tensor_libTensor)r'rs r get_classesr-ash   wLL))  FM66 7  =G=N=NO) )sAA0 c tj|tj|Dcgc]>}tj|r%t j |tjn|@c}}|Scc}w)zSerializes many sparse tensors into a batch. Args: tensors: a tensor structure to serialize. Returns: `tensors` with any sparse tensors replaced by the serialized batch. out_type) rrrr is_sparserserialize_many_sparserrr'rrs rserialize_many_sparse_tensorsr4ssn gLL)(    (&&vG.45( # * (sAA4 c tj|tj|Dcgc]C}t|tj r%t j|tjn|Ec}}|Scc}w)zSerializes sparse tensors. Args: tensors: a tensor structure to serialize. Returns: `tensors` with any sparse tensors replaced by their serialized version. r/) rrrr*rr rserialize_sparserrr3s rserialize_sparse_tensorsr7sq gLL)(  FM66 7!!&6>>B=CD( # * (sAA9 N)__doc__tensorflow.python.data.utilrtensorflow.python.frameworkrrrr+rtensorflow.python.opsrrrr!r(r-r4r7rrr=s@6,.5<4, M & & .$ $ r