o 2h @spdZddlZddlmZddlmZddlmZddlmZddlm Z dZ e d Gd d d ej ej d Z dS) z?Utilities for including Python state in TensorFlow checkpoints.N) constant_op)dtypes)ops)base) tf_exportpy_stateztrain.experimental.PythonStatec@s4eZdZdZejddZejddZddZdS) PythonStateaA mixin for putting Python state in an object-based checkpoint. This is an abstract class which allows extensions to TensorFlow's object-based checkpointing (see `tf.train.Checkpoint`). For example a wrapper for NumPy arrays: ```python import io import numpy class NumpyWrapper(tf.train.experimental.PythonState): def __init__(self, array): self.array = array def serialize(self): string_file = io.BytesIO() try: numpy.save(string_file, self.array, allow_pickle=False) serialized = string_file.getvalue() finally: string_file.close() return serialized def deserialize(self, string_value): string_file = io.BytesIO(string_value) try: self.array = numpy.load(string_file, allow_pickle=False) finally: string_file.close() ``` Instances of `NumpyWrapper` are checkpointable objects, and will be saved and restored from checkpoints along with TensorFlow state like variables. ```python root = tf.train.Checkpoint(numpy=NumpyWrapper(numpy.array([1.]))) save_path = root.save(prefix) root.numpy.array *= 2. assert [2.] == root.numpy.array root.restore(save_path) assert [1.] == root.numpy.array ``` cCdS)z3Callback to serialize the object. Returns a string.N)selfr r c/var/www/html/chatgem/venv/lib/python3.10/site-packages/tensorflow/python/trackable/python_state.py serializeKzPythonState.serializecCr )z#Callback to deserialize the object.Nr )r string_valuer r r deserializeOrzPythonState.deserializecCsJttj|tjd}Wdt|iS1swYt|iS)z+Implements Trackable._serialize_to_tensors.)dtypeN)r init_scoperconstantr rstring PYTHON_STATE)r valuer r r _serialize_to_tensorsSs  z!PythonState._serialize_to_tensorsN) __name__ __module__ __qualname____doc__abcabstractmethodr rrr r r r rs-   r) metaclass)rrtensorflow.python.frameworkrrrtensorflow.python.trackabler tensorflow.python.util.tf_exportrr TrackableABCMetarr r r r s