|j/4ddlmZddlmZddlZddlZddlmZddlm Z m Z ddl m Z m Z ddlmZmZdd lmZdd lmZmZd d lmZmZerdd lmZgZddZddZe ddgiddZe ddgddZddZdddZ dddZ!dS) ) annotations) TYPE_CHECKINGN)_C_ops)ParamAliasDecoratorparam_one_alias) check_typecheck_variable_and_dtype)in_dynamic_or_pir_mode use_pir_api)Variable) LayerHelpercore)_complex_to_real_dtypeassign)Tensorinputrreturnct|dttjjfdt |j}ttj |d}|S)a% Returns the number of dimensions for a tensor, which is a 0-D int32 Tensor. Args: input (Tensor): The input Tensor with shape of :math:`[N_1, N_2, ..., N_k]`, the data type is arbitrary. Returns: Tensor, the output data type is int32.: The 0-D tensor with the dimensions of the input Tensor. Examples: .. code-block:: python >>> import paddle >>> input = paddle.rand((3, 100, 100)) >>> rank = paddle.rank(input) >>> print(rank.numpy()) 3 rint32) r r paddlepirValuelenshapernparray)rndimsouts g/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/paddle/tensor/attribute.pyrankr"%sO*ug&**:;WEEE   E %)) * *C Jc(trtj|}d|_|St |dgddt d it }|d}|dd|id|id d|_|S) a Get the shape of the input. .. code-block:: text Case1: Given N-D Tensor: input = [ [1, 2, 3, 4], [5, 6, 7, 8] ] Then: input.shape = [2, 4] Case2: Given SelectedRows: input.rows = [0, 4, 19] input.height = 20 input.value = [ [1, 2], [3, 4], [5, 6] ] # inner tensor Then: input.shape = [3, 2] Args: input (Tensor): The input can be N-D Tensor or SelectedRows with data type bool, bfloat16, float16, float32, float64, int32, int64. If input variable is type of SelectedRows, returns the shape of it's inner tensor. Returns: Tensor: The shape of the input variable. Examples: .. code-block:: pycon >>> import numpy as np >>> import paddle >>> paddle.enable_static() >>> inputs = paddle.static.data(name="x", shape=[3, 100, 100], dtype="float32") >>> output = paddle.shape(inputs) >>> exe = paddle.static.Executor(paddle.CPUPlace()) >>> exe.run(paddle.static.default_startup_program()) >>> img = np.ones((3, 100, 100)).astype(np.float32) >>> res = exe.run(paddle.static.default_main_program(), feed={'x': img}, fetch_list=[output]) >>> print(res) [array([ 3, 100, 100], dtype=int64)] Tr) booluint16float16float32float64rint64 complex64 complex128r& float8_e4m3fn float8_e5m2rrdtypeInputOut)typeinputsoutputs stop_gradientN)r) r rshape64r6r rlocals"create_variable_for_type_inference append_op)rr helpers r!rrAs^!nU##        #   &111177g7FFU#CL    ! r#xr%ct|tjtjjtjjfstdt||j }|tj j j kpC|tj j jkp)|tjj kp|tjjk}|S)amReturn whether x is a tensor of complex data type(complex64 or complex128). .. note:: Alias Support: The parameter name ``input`` can be used as an alias for ``x``. For example, ``input=tensor_x`` is equivalent to ``x=tensor_x``. Args: x (Tensor): The input tensor. input: An alias for ``x`` , with identical behavior. Returns: bool: True if the data type of the input is complex data type, otherwise false. Examples: .. code-block:: python >>> import paddle >>> x = paddle.to_tensor([1 + 2j, 3 + 4j]) >>> print(paddle.is_complex(x)) True >>> x = paddle.to_tensor([1.1, 1.2]) >>> print(paddle.is_complex(x)) False >>> x = paddle.to_tensor([1, 2, 3]) >>> print(paddle.is_complex(x)) False )Expected Tensor, but received type of x: ) isinstancerrstaticr rr TypeErrorr3r0rVarDescVarType COMPLEX64 COMPLEX128DataType)r<r0is_complex_dtypes r! is_complexrHsB  FM6=16:3C D  OMDGGMMNNN GE %// - DL(3 3 - DM+ + - DM, ,  r#c@t|tjtjjtjjfstdt||j }|tj j j kp|tj j jkp|tj j jkpm|tj j jkpS|tjjkp>|tjjkp)|tjjkp|tjjk}|S)a2 Returns whether the dtype of `x` is one of paddle.float64, paddle.float32, paddle.float16, and paddle.bfloat16. .. note:: Alias Support: The parameter name ``input`` can be used as an alias for ``x``. For example, ``is_floating_point(input=tensor_x)`` is equivalent to ``is_floating_point(x=tensor_x)``. Args: x (Tensor): The input tensor. alias: ``input``. Returns: bool: True if the dtype of `x` is floating type, otherwise false. Examples: .. code-block:: python >>> import paddle >>> x = paddle.arange(1., 5., dtype='float32') >>> y = paddle.arange(1, 5, dtype='int32') >>> print(paddle.is_floating_point(x)) True >>> print(paddle.is_floating_point(y)) False r>)r?rrr@r rrrAr3r0rrBrCFP32FP64FP16BF16rFFLOAT32FLOAT64FLOAT16BFLOAT16)r<r0 is_fp_dtypes r!is_floating_pointrSs6  FM6=16:3C D  OMDGGMMNNN GE %** + DL(- - + DL(- - + DL(- - + DM) )  + DM) )  + DM) )  + DM* * r#ct|tjtjjtjjfstdt||j }d}ts|tj j jkpg|tj j jkpM|tj j jkp3|tj j jkp|tj j jk}ni|tjjkpS|tjjkp>|tjjkp)|tjjkp|tjjk}|S)apReturn whether x is a tensor of integral data type. Args: x (Tensor): The input tensor. Returns: bool: True if the data type of the input is integer data type, otherwise false. Examples: .. code-block:: python >>> import paddle >>> x = paddle.to_tensor([1 + 2j, 3 + 4j]) >>> print(paddle.is_integer(x)) False >>> x = paddle.to_tensor([1.1, 1.2]) >>> print(paddle.is_integer(x)) False >>> x = paddle.to_tensor([1, 2, 3]) >>> print(paddle.is_integer(x)) True r>F)r?rrr@r rrrAr3r0r rrBrCUINT8INT8INT16INT32INT64rF)r<r0 is_int_dtypes r! is_integerr[s=4  FM6=16:3C D  OMDGGMMNNN GEL == T\)/ / 3 ,11 3 ,22 3 ,22 3 ,22   T]( ( , ** , ++ , ++ , ++  r#name str | NonecDtrtj|St|dddgdt d it }|t|}| dd|id|i|S) a Returns a new Tensor containing real values of the input Tensor. Args: x (Tensor): the input Tensor, its data type could be complex64 or complex128. name (str|None, optional): The default value is None. Normally there is no need for user to set this property. For more information, please refer to :ref:`api_guide_Name` . Returns: Tensor: a Tensor containing real values of the input Tensor. Examples: .. code-block:: python >>> import paddle >>> x = paddle.to_tensor( ... [[1 + 6j, 2 + 5j, 3 + 4j], [4 + 3j, 5 + 2j, 6 + 1j]]) >>> print(x) Tensor(shape=[2, 3], dtype=complex64, place=Place(cpu), stop_gradient=True, [[(1+6j), (2+5j), (3+4j)], [(4+3j), (5+2j), (6+1j)]]) >>> real_res = paddle.real(x) >>> print(real_res) Tensor(shape=[2, 3], dtype=float32, place=Place(cpu), stop_gradient=True, [[1., 2., 3.], [4., 5., 6.]]) >>> real_t = x.real() >>> print(real_t) Tensor(shape=[2, 3], dtype=float32, place=Place(cpu), stop_gradient=True, [[1., 2., 3.], [4., 5., 6.]]) r<r+r,realr/Xr2r3r4r5N)r_) r rr_r rr8r9r input_dtyper:r<r\r;r s r!r_r_%H {1~~ C+|)DfMMM00vxx0077(););)=)=>>8   fc1Xs|LLL r#cDtrtj|St|dddgdt d it }|t|}| dd|id|i|S) a Returns a new tensor containing imaginary values of input tensor. Args: x (Tensor): the input tensor, its data type could be complex64 or complex128. name (str|None, optional): The default value is None. Normally there is no need for user to set this property. For more information, please refer to :ref:`api_guide_Name` . Returns: Tensor: a tensor containing imaginary values of the input tensor. Examples: .. code-block:: python >>> import paddle >>> x = paddle.to_tensor( ... [[1 + 6j, 2 + 5j, 3 + 4j], [4 + 3j, 5 + 2j, 6 + 1j]]) >>> print(x) Tensor(shape=[2, 3], dtype=complex64, place=Place(cpu), stop_gradient=True, [[(1+6j), (2+5j), (3+4j)], [(4+3j), (5+2j), (6+1j)]]) >>> imag_res = paddle.imag(x) >>> print(imag_res) Tensor(shape=[2, 3], dtype=float32, place=Place(cpu), stop_gradient=True, [[6., 5., 4.], [3., 2., 1.]]) >>> imag_t = x.imag() >>> print(imag_t) Tensor(shape=[2, 3], dtype=float32, place=Place(cpu), stop_gradient=True, [[6., 5., 4.], [3., 2., 1.]]) r<r+r,imagr/r`r2raN)rf) r rrfr rr8r9rrbr:rcs r!rfrfUrdr#)rrrr)r<rrr%)N)r<rr\r]rr)" __future__rtypingrnumpyrrrpaddle.utils.decorator_utilsrrbase.data_feederr r base.frameworkr r common_ops_importr frameworkrrcreationrrr__all__r"rrHrSr[r_rfr#r!rrs#"""""  MMMMMMMMCCCCCCCC@@@@@@@@(((((())))))))44444444 8PPPPfcG9%&&+++'&+\#w  )))! )X2222j-----`-------r#