o #j)@sddlZddlmZddlmZddlmZmZddl m Z m Z ddl m Z ddlmZejjejejjejejjejejjejejjejejjejejjejejjejejjejejjejejjejejj ej i Z!e"d eje"d eje"d eje"d eje"d eje"deje"deje"deje"deje"deje"deje"dej i Z#ddZ$ea%et%ea&ddZ'ddZ(ddZ)ddZ*ed'dd Z+Gd!d"d"Z, d'd#d$Z-d%d&Z.dS)(N)VarDesc)DataType)Programset_global_program) parameter set_parameter) unique_name)signature_safe_contextmanagerZfloat32Zfloat64Zfloat16Zint32Zint16Zint64Zbool_uint16Zuint8Zint8Z complex64Z complex128cCsHt|tr|dkrtd}nt|}|tvrt|Std|)z Convert the data type in numpy to the data type in Paddle. Args: np_dtype (np.dtype|str): The data type in numpy or valid data type string. Returns: core.DataType : The data type in Paddle. Zbfloat16r zNot supported numpy dtype %s) isinstancestrnpdtypenp_type_to_paddle_typekeys ValueError)Znp_dtyperrP/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/pir/core.pyconvert_np_dtype_to_dtype_9s     rcCtS)a Get default/global startup program. The :code:`paddle.nn` function will append the initialization operators into startup program. The :code:`startup_program` will initialize the parameters by the OPs. This method will return the default or the current startup program. Users can use :ref:`api_paddle_ir_core_program_guard` to switch :ref:`api_paddle_ir_Program` . Returns: Program: current default startup program. Returns type: Examples: .. code-block:: python >>> import paddle >>> paddle.enable_static() >>> x = paddle.static.data(name="x", shape=[-1, 784], dtype='float32') >>> out = paddle.static.nn.fc(name="fc", x=x, size=10, activation="relu") >>> print("main program is: {}".format(paddle.static.default_main_program())) >>> print("start up program is: {}".format(paddle.static.default_startup_program())) _startup_program_rrrrdefault_startup_program[srcCr)a7 This API can be used to get ``default main program`` which store the descriptions of Ops and tensors. For example ``z = paddle.add(x, y)`` will create a new ``add`` Op and a new ``z`` tensor, and they will be recorded in ``default main program`` . The ``default main program`` is the default value for ``Program`` parameter in a lot of APIs. For example, the :code:`Executor.run()` will execute the :code:`default_main_program` when the program is not specified. If you want to switch the ``default main program``, you can use :ref:`api_paddle_ir_core_program_guard` . Returns: Program: A ``Program`` which holding the descriptions of OPs and tensors in the network. Examples: .. code-block:: python >>> import paddle >>> paddle.enable_static() >>> # Sample Network: >>> x = paddle.static.data(name='x', shape=[100, 100], dtype='float32') >>> y = paddle.static.data(name='y', shape=[100, 100], dtype='float32') >>> out = paddle.add(x, y) >>> print the number of blocks in the program, 1 in this case >>> print(paddle.static.default_main_program().num_blocks) # 1 >>> print the default_main_program >>> print(paddle.static.default_main_program()) )_main_program_rrrrdefault_main_programxs!rcCst}|att|S)z Switch the main program to a new program. Args: program(Program): The new main program Returns: Program: The previous main program )rrprogramZ prev_programrrrswitch_main_programs rcCs t}|a|S)z Switch the startup program to a new program Args: program(Program): The new startup program Returns: Program: The previous startup program rrrrrswitch_startup_programs rccsddlm}||dtdt|}|dur!||dtdt|}zdVWt||dur4t|dSdSt||durCt|ww)a :api_attr: Static Graph Change the global main program and startup program with ``with`` statement. Layer functions in the Python ``with`` block will append operators and Tensors to the new main programs. Args: main_program(Program): New main program inside ``with`` statement. startup_program(Program, optional): New startup program inside ``with`` statement. :code:`None` means not changing startup program, default_startup_program is still used. Default: None. Examples: .. code-block:: python :name: code-example-1 >>> import paddle >>> paddle.enable_static() >>> main_program = paddle.static.Program() >>> startup_program = paddle.static.Program() >>> with paddle.static.program_guard(main_program, startup_program): ... data = paddle.static.data(name='image', shape=[None, 784, 784], dtype='float32') ... hidden = paddle.static.nn.fc(x=data, size=10, activation='relu') Notes: The temporary :code:`Program` can be used if the user does not need to construct either of startup program or main program. Examples: .. code-block:: python :name: code-example-2 >>> import paddle >>> paddle.enable_static() >>> main_program = paddle.static.Program() >>> # does not care about startup program. Just pass a temporary value. >>> with paddle.static.program_guard(main_program, paddle.static.Program()): ... data = paddle.static.data(name='image', shape=[None, 784, 784], dtype='float32') r) check_type main_programzpaddle.static.program_guardNstartup_program)Zbase.data_feederr rrr)r!r"r rrr program_guards0 ,  r#c@seZdZddZdS) ParameterMetacCs||_||_dSN)shaper)selfr&rrrr__init__s zParameterMeta.__init__N)__name__ __module__ __qualname__r(rrrrr$s r$c Ksd|vrtd|durt|tst|}|}|std}t}t}t||}t ||d}||| } d| _ t | |Wdn1sKwY| |t tt|||} |dd} | | _d| _ Wd| S1sywY| S)N initializerzRinitializer is None, if you want to create parameter, please pass its initializer.rT trainable)rr rrr generaterrr$r#Z global_blockZ persistablerZmove_parameters_fromrgetZ stop_gradient) rr&namekwargsZop_result_namer"r!Zparameter_metar,Z init_resultparamr-rrrcreate_parameters>          r3cCs|ddl}ddlm}m}ddlm}t||jjr<|j |j }d}|dur0t||j s.J|S| |jj|}|S|S)z' Convert Tensor into OpResult. rN)core framework)_global_parameter_recorderT)paddleZ paddle.baser4r5Z+paddle.jit.pir_dy2static.parameter_recorderr6r eagerZTensorblockZ_find_var_recursiver0Variabler/Zpirr)Ztensorr7r4r5r6Znew_varZis_persistablerrr_convert_into_opresult*s  r;r%)/numpyrZpaddle.base.corerZpaddle.base.libpaddlerZpaddle.base.libpaddle.pirrrZ_pir_opsrrbaser Zbase.wrapped_decoratorr ZVarTypeZFP32ZFLOAT32ZFP64ZFLOAT64ZFP16ZFLOAT16ZBF16ZBFLOAT16ZINT32ZINT16ZINT64ZBOOLZUINT8ZINT8Z COMPLEX64Z COMPLEX128Zvartype_to_datatyperrrrrrrrrr#r$r3r;rrrrs^                            $ A &