B ٻdD@sdZddlmZddlmZddlmZddlmZddlmZddlm Z dd Z d d Z d d Z ddZ ddZddZddZdS)z/Python dataset sparse tensor utility functions.)nest)dtypes)ops) sparse_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|]}|tjkVqdS)N)r SparseTensor).0cr T/var/www/html/venv/lib/python3.7/site-packages/tensorflow/python/data/util/sparse.py !szany_sparse..)anyrflatten)classesr r r any_sparses rc Cs,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|]\}}|tjkrtn|qSr )rrrZ unknown_shape)r shaper r r r 1sz#as_dense_shapes..)rpack_sequence_aszipr)shapesrretr r r as_dense_shapes$s rc Cs,t|ddtt|t|D}|S)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|]\}}|tjkrtjn|qSr )rrrvariant)r tyr r r r rDsz"as_dense_types..)rrrr)typesrrr r r as_dense_types7s rc 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|],\}}}}|tjkr,tj|||jdn|qS))ZdtypeZrank)rrrZdeserialize_sparseZndims)r tensorrrr r r r rXsz.deserialize_sparse_tensors..)rrrr)tensorsrrrrr r r deserialize_sparse_tensorsJs rcCst|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|tjrtjntjqSr ) isinstancerrrZTensor)r rr r r rmszget_classes..)rrr)rr r r get_classesas r!cCs 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|r"tj|tjdn|qS))out_type)rZ is_sparserZserialize_many_sparserr)r rr r r r~sz1serialize_many_sparse_tensors..)rrr)rrr r r serialize_many_sparse_tensorsss r#cCs t|ddt|D}|S)zSerializes sparse tensors. Args: tensors: a tensor structure to serialize. Returns: `tensors` with any sparse tensors replaced by their serialized version. cSs,g|]$}t|tjr$tj|tjdn|qS))r")r rrrZserialize_sparserr)r rr r r rsz,serialize_sparse_tensors..)rrr)rrr r r serialize_sparse_tensorss r$N)__doc__Ztensorflow.python.data.utilrZtensorflow.python.frameworkrrrrZtensorflow.python.opsrrrrrr!r#r$r r r r s