o i eD#@sdZddlZddlZddlmmZddlmZddlm Z ddl m Z ddl m ZddlmZddlmZdd lmZdd lmZdd lmZead d ZeetjeddddZeddddZ edddZ!dS)z2Keras initializer serialization / deserialization.N) initializers)initializers_v1)serialization_lib) serialization) generic_utils) tf_inspect)tf2)init_ops) keras_exportcsvttds it_dt_tjrtjtjjkrdSit_tjjt_tj tjd<tj tjd<tj tjd<tj tjd<tj tjd<tjtjd<tjtjd <tjtjd <tjtjd <tjtjd <tjtjd <tjtjd<tjtjd<tjtjd<tjtjd<tj tjd<tj tjd<tj tjd<tj tjd<tjtjd<tjtjd<tjjri}tjtj|tgfddd|D]\}}|tj|<|tjt|<qnItjjjtjjj tjjj!tjjjj"tjjj#tjjj$tjjj%tjjj&t'j t'j t'jt'jt'jt'jt'jd}|D]\}}|tj|<|tjt|<qtjdtjd<tjdtjd<tjd tjd!<tjd"tjd#<dS)$z;Populates dict ALL_OBJECTS with every built-in initializer. ALL_OBJECTSN ConstantV2GlorotNormalV2GlorotUniformV2 HeNormalV2 HeUniformV2 IdentityV2 LecunNormalV2LecunUniformV2OnesV2 OrthogonalV2RandomNormalV2RandomUniformV2TruncatedNormalV2VarianceScalingV2ZerosV2glorot_normalV2glorot_uniformV2 he_normalV2 he_uniformV2lecun_normalV2lecun_uniformV2cst|o t|S)N)inspectisclass issubclass)xbase_clsN/var/www/myenv/lib/python3.10/site-packages/keras/src/initializers/__init__.py\sz1populate_deserializable_objects..) obj_filter)Constant GlorotNormal GlorotUniformIdentityOnes OrthogonalVarianceScalingZerosHeNormal HeUniform LecunNormal LecunUniform RandomNormal RandomUniformTruncatedNormal random_normalnormalrandom_uniformuniformonesonezeroszero)(hasattrLOCALr GENERATED_WITH_V2tf __internal__renabledrr+r,r-r3r4r.r5r6r/r0r7r8r9r1r2 Initializerr!populate_dict_with_module_objectsitems to_snake_casecompatv1constant_initializerglorot_normal_initializerglorot_uniform_initializeridentityones_initializerorthogonal_initializervariance_scaling_initializerzeros_initializerr)v2_objskeyvaluev1_objsr'r%r(populate_deserializable_objects's                           rZzkeras.initializers.serializeFcCsTt|dur dSt|ttjstdt|d|r%t |St |S)NzThe `keras.initializers.serialize()` API should only be used for objects of type `keras.initializers.Initializer`. Found an instance of type z+, which may lead to improper serialization.) rZ isinstancetuplerCr valueswarningswarntypelegacy_serializationserialize_keras_objectr) initializeruse_legacy_formatr'r'r( serializes  rezkeras.initializers.deserializecCs2t|rtj|tj|ddStj|tj|ddS)z/Return an `Initializer` object from its config.rc)module_objectscustom_objectsprintable_module_name)rZradeserialize_keras_objectrCr r)configrgrdr'r'r( deserializesrkzkeras.initializers.getcCsv|durdSt|trd|v}t||dSt|tr%t|id}t|St|r3t|r1|}|Stdt|)a)Retrieve a Keras initializer by the identifier. The `identifier` may be the string name of a initializers function or class (case-sensitively). >>> identifier = 'Ones' >>> tf.keras.initializers.deserialize(identifier) <...keras.initializers.initializers.Ones...> You can also specify `config` of the initializer to this function by passing dict containing `class_name` and `config` as an identifier. Also note that the `class_name` must map to a `Initializer` class. >>> cfg = {'class_name': 'Ones', 'config': {}} >>> tf.keras.initializers.deserialize(cfg) <...keras.initializers.initializers.Ones...> In the case that the `identifier` is a class, this method will return a new instance of the class by its constructor. Args: identifier: String or dict that contains the initializer name or configurations. Returns: Initializer instance base on the input identifier. Raises: ValueError: If the input identifier is not a supported type or in a bad format. Nmodule)rd) class_namerjz,Could not interpret initializer identifier: ) r[dictrkstrgetcallabler!r" ValueError) identifierrdrjr'r'r(rps"     rp)F)NF)"__doc__ threadingr^tensorflow.compat.v2rLv2rEkeras.src.initializersrrkeras.src.savingrkeras.src.saving.legacyrrakeras.src.utilsrrr!tensorflow.pythonrtensorflow.python.opsr tensorflow.python.util.tf_exportr localrCrZglobalsupdater rerkrpr'r'r'r(s.         ]