o ?e\@sdZddlmZddlmZddlmZddlmZddlmZddl m Z dd Z d d Z d d Z ddZddZddZddZdS)z/Python dataset sparse tensor utility functions.)nest)dtypes) sparse_tensor)tensor) tensor_shape) sparse_opscCstddt|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. css|]}|tjuVqdS)N)r SparseTensor).0cr c/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/data/util/sparse.py !szany_sparse..)anyrflatten)classesr r r any_sparses rc C,t|ddtt|t|D}|S)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 cSs&g|]\}}|tjurtn|qSr )rrrZ unknown_shape)r shaper r r r 0sz#as_dense_shapes..rpack_sequence_aszipr)shapesrretr r r as_dense_shapes$ rc Cr)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 cSs$g|]\}}|tjurtjn|qSr )rrrvariant)r tyr r r r rCsz"as_dense_types..r)typesrrr r r as_dense_types7rrc Cs<t|ddtt|t|t|t|D}|S)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. cSs4g|]\}}}}|tjurtj|||jdn|qS))ZdtypeZrank)rrrZdeserialize_sparseZndims)r rrrr r r r rWs  z.deserialize_sparse_tensors..r)tensorsrrrrr r r deserialize_sparse_tensorsJs r!cCst|ddt|DS)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. cSs$g|]}t|tjr tjntjqSr ) isinstancerr tensor_libZTensorr rr r r rls  zget_classes..rrr)r r r r get_classesas r&cC t|ddt|D}|S)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. cSs*g|]}t|rtj|tjdn|qS)Zout_type)rZ is_sparserZserialize_many_sparserrr$r r r r}s z1serialize_many_sparse_tensors..r%r rr r r serialize_many_sparse_tensorss r*cCr')zSerializes sparse tensors. Args: tensors: a tensor structure to serialize. Returns: `tensors` with any sparse tensors replaced by their serialized version. cSs,g|]}t|tjrtj|tjdn|qSr()r"rrrZserialize_sparserrr$r r r rs  z,serialize_sparse_tensors..r%r)r r r serialize_sparse_tensorsr+r,N)__doc__Ztensorflow.python.data.utilrZtensorflow.python.frameworkrrrr#rZtensorflow.python.opsrrrrr!r&r*r,r r r r s