\;j -ddlZddlZddlmZddlmZmZddlmZm Z ddl m Z ddl m Z mZdd lmZmZgZd Zd Zd Zd ZdZddZddZy)N)_C_ops) check_typecheck_variable_and_dtype)in_dynamic_or_pir_mode in_pir_mode)Variable) LayerHelpercore)_complex_to_real_dtypeassignct|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 inputint32) rr paddlepirValuelenshapernparray)rndimsouts `G:\00. PROJECTS\API\Inventory\templateJSON\kerjaOCR\Lib\site-packages\paddle/tensor/attribute.pyrankrsI*ug&***:*:;WE  E %) *C Jctrtj|}d|_|St |dgddt d it }|jd}|jdd|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 (Variable): 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: Variable (Tensor): The shape of the input variable. Examples: .. code-block:: python >>> 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=int32)] Tr) booluint16float16float32float64rint64 complex64 complex128r rrdtypeInputOut)typeinputsoutputs stop_gradient)r) rrrr.rr locals"create_variable_for_type_inference append_op)rrhelpers rrr;s^ll5!       "1177g7FU#CL  ! rct|tjtjjtj j fstdt||j}|tjjjk(xse|tjjjk(xs<|tjjk(xs|tjjk(}|S)aReturn whether x is a tensor of complex data type(complex64 or complex128). Args: x (Tensor): The input tensor. 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: ) isinstancerTensorstaticr rr TypeErrorr+r(r VarDescVarType COMPLEX64 COMPLEX128DataType)xr(is_complex_dtypes r is_complexr@s4  FMM6==116::3C3C D CDG9MNN GGE %%/// - DLL((33 3 - DMM++ + - DMM,, ,  rct|tjtjjfst dt ||j}|tjjjk(xsy|tjjjk(xsP|tjjjk(xs'|tjjjk(}|S)aK Returns whether the dtype of `x` is one of paddle.float64, paddle.float32, paddle.float16, and paddle.bfloat16. Args: x (Tensor): The input tensor. 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 r4)r5rr6r7r r8r+r(r r9r:FP32FP64FP16BF16)r>r( is_fp_dtypes ris_floating_pointrGs, a&--)?)?@ ACDG9MNN GGE %%*** . DLL((-- - . DLL((-- - . DLL((-- -  rct|tjtjjtj j fstdt||j}d}ts|tjjjk(xs|tjjjk(xsy|tjjj k(xsP|tjjj"k(xs'|tjjj$k(}|S|tj&jk(xsz|tj&jk(xs[|tj&j k(xs<|tj&j"k(xs|tj&j$k(}|S)aqReturn whether x is a tensor of integeral 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 r4F)r5rr6r7r rOpResultr8r+r(rr r9r:UINT8INT8INT16INT32INT64r=)r>r( is_int_dtypes r is_integerrPsz4  FMM6==116::3F3F G CDG9MNN GGEL = T\\))// / 3 ,,111 3 ,,222 3 ,,222 3 ,,222   T]]'' ' , *** , +++ , +++ , +++  rc trtj|St|dddgdt d it }|j t|j}|jdd|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, 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'Xr*r+r,r-)rR) rrrRrr r/r0r input_dtyper1r>namer2rs rrRrR H{{1~ C+|)DfM0vx077(););)=>8  fc1Xs|L rc trtj|St|dddgdt d it }|j t|j}|jdd|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, 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'rSr*rT)rZ) rrrZrr r/r0r rUr1rVs rrZrZ;rXr)N)numpyrrrbase.data_feederrrbase.frameworkrrcommon_ops_importr frameworkr r creationr r__all__rrr@rGrPrRrZrrrcsM" C@()4 8Nb%PD2j-`-r