o Me[@sdZddlZddlmmZddlmZddlmZddl m Z ddl m Z ddl m ZddlmZdd lmZdd lmZead d Zeetjed ddZeddddZedddZdS)z2Keras initializer serialization / deserialization.N)initializers_v1)initializers_v2) 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!tjjj"j#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_OBJECTSNZ ConstantV2ZGlorotNormalV2ZGlorotUniformV2Z HeNormalV2Z HeUniformV2Z IdentityV2Z LecunNormalV2ZLecunUniformV2ZOnesV2Z OrthogonalV2ZRandomNormalV2ZRandomUniformV2ZTruncatedNormalV2ZVarianceScalingV2ZZerosV2Zglorot_normalV2Zglorot_uniformV2Z he_normalV2Z he_uniformV2Zlecun_normalV2Zlecun_uniformV2cst|o t|SN)inspectisclass issubclass)xZbase_clsKD:\Projects\ConvertPro\env\Lib\site-packages\keras/initializers/__init__.pyZsz1populate_deserializable_objects..)Z obj_filter)Constant GlorotNormal GlorotUniformIdentityOnes OrthogonalVarianceScalingZerosHeNormal HeUniform LecunNormal LecunUniform RandomNormal RandomUniformTruncatedNormalZ random_normalnormalZrandom_uniformuniformZonesoneZzeroszero))hasattrLOCALr ZGENERATED_WITH_V2tfZ __internal__renabledrrrrrrrrrrrr r!r"rrZ InitializerrZ!populate_dict_with_module_objectsitemsZ to_snake_casecompatv1Zconstant_initializerZglorot_normal_initializerZglorot_uniform_initializerZ initializersidentityZones_initializerZorthogonal_initializerZvariance_scaling_initializerZzeros_initializerr)Zv2_objskeyvalueZv1_objsrrrpopulate_deserializable_objects%s                           r1zkeras.initializers.serializecCs t|Sr )rZserialize_keras_object) initializerrrr serializes r3zkeras.initializers.deserializecCsttj|tj|ddS)z/Return an `Initializer` object from its config.r2)Zmodule_objectscustom_objectsZprintable_module_name)r1rZdeserialize_keras_objectr(r )configr4rrr deserializesr6zkeras.initializers.getcCsd|durdSt|trt|St|trt|}t|St|r*t|r(|}|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_v2.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_v2.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. Nz,Could not interpret initializer identifier: ) isinstancedictr6strcallabler r ValueError) identifierrrrgets"    r=r )__doc__ threadingZtensorflow.compat.v2r,v2r)Zkeras.initializersrrZkeras.saving.legacyrZ keras.utilsrrr Ztensorflow.pythonrZtensorflow.python.opsrZ tensorflow.python.util.tf_exportr localr(r1globalsupdater r3r6r=rrrrs*        ]