\;j/hddlZddlmZmZddlmZddlmZddlm Z m Z m Z gZ dZ dZd d Zd d Zy) N) check_typecheck_variable_and_dtype) in_pir_mode)Variable) LayerHelpercorein_dynamic_modecRtr"t|tsJdt|St r^t|t j jr|js tdt jj|St|tr1|jtjj j"k7r tdt%d it'}|j)d}d|_|j-dd|gid|gi |S) a5 This OP is used to get the length of the input array. Args: array (list|Tensor): The input array that will be used to compute the length. In dynamic mode, ``array`` is a Python list. But in static graph mode, array is a Tensor whose VarType is LOD_TENSOR_ARRAY. Returns: Tensor: 1-D Tensor with shape [1], which is the length of array. Examples: .. code-block:: python >>> import paddle >>> arr = paddle.tensor.create_array(dtype='float32') >>> x = paddle.full(shape=[3, 3], fill_value=5, dtype="float32") >>> i = paddle.zeros(shape=[1], dtype="int32") >>> arr = paddle.tensor.array_write(x, i, array=arr) >>> arr_len = paddle.tensor.array_length(arr) >>> print(arr_len) 1 9The 'array' in array_write must be a list in dygraph mode8array should be tensor array vairable in array_length Opint64dtypeTlod_array_lengthXOuttypeinputsoutputs) array_length)r isinstancelistlenrpaddlepirOpResultis_dense_tensor_array_type TypeError_pir_opsrrrr VarDescVarTypeLOD_TENSOR_ARRAYrlocals"create_variable_for_type_inference stop_gradient append_op)arrayhelpertmps \G:\00. PROJECTS\API\Inventory\templateJSON\kerjaOCR\Lib\site-packages\paddle/tensor/array.pyrrs2 4  G F G 5z 5&**"5"56335J ++E225(+zzT\\11BBBJ 8vx877g7F #%>SEN   ctr[t|tsJdt|tsJd|jdgk(sJd|j d}||St r_t|tjjr|js tdtjj||St|ddgd tdit!}t|tr1|j"t$j&j(j*k7r td |j-|j. }|j1d |g|gd d|gi|S)a This OP is used to read data at the specified position from the input array. Case: .. code-block:: text Input: The shape of first three tensors are [1], and that of the last one is [1,2]: array = ([0.6], [0.1], [0.3], [0.4, 0.2]) And: i = [3] Output: output = [0.4, 0.2] Args: array (list|Tensor): The input array. In dynamic mode, ``array`` is a Python list. But in static graph mode, array is a Tensor whose ``VarType`` is ``LOD_TENSOR_ARRAY``. i (Tensor): 1-D Tensor, whose shape is [1] and dtype is int64. It represents the specified read position of ``array``. Returns: Tensor: A Tensor that is read at the specified position of ``array``. Examples: .. code-block:: python >>> import paddle >>> arr = paddle.tensor.create_array(dtype="float32") >>> x = paddle.full(shape=[1, 3], fill_value=5, dtype="float32") >>> i = paddle.zeros(shape=[1], dtype="int32") >>> arr = paddle.tensor.array_write(x, i, array=arr) >>> item = paddle.tensor.array_read(arr, i) >>> print(item.numpy()) [[5. 5. 5.]] z6The 'array' in array_read must be list in dygraph modeztrt|tsJdt|tsJd|jdgk(sJd|j d}|t |j }t|tsJd|t|ksJd|t|kr|||<|S|j||Strt|dd gd t|tjjstd t!|d |?t|tjjr|j#s td |)tj$j |j }tj$j'|||}|St|dd gd t)|dtd t+dit-}|Lt|tr1|j t.j0j2j4k7r td|M|j7|j8dt.j0j2j4|j }|j;d|g|gdd|gi|S)a> This OP writes the input ``x`` into the i-th position of the ``array`` returns the modified array. If ``array`` is none, a new array will be created and returned. Args: x (Tensor): The input data to be written into array. It's multi-dimensional Tensor or LoDTensor. Data type: float32, float64, int32, int64 and bool. i (Tensor): 1-D Tensor with shape [1], which represents the position into which ``x`` is written. array (list|Tensor, optional): The array into which ``x`` is written. The default value is None, when a new array will be created and returned as a result. In dynamic mode, ``array`` is a Python list. But in static graph mode, array is a Tensor whose ``VarType`` is ``LOD_TENSOR_ARRAY``. Returns: list|Tensor: The input ``array`` after ``x`` is written into. Examples: .. code-block:: python >>> import paddle >>> arr = paddle.tensor.create_array(dtype="float32") >>> x = paddle.full(shape=[1, 3], fill_value=5, dtype="float32") >>> i = paddle.zeros(shape=[1], dtype="int32") >>> arr = paddle.tensor.array_write(x, i, array=arr) >>> item = paddle.tensor.array_read(arr, i) >>> print(item.numpy()) [[5. 5. 5.]] zBThe input data 'x' in array_write must be Variable in dygraph modez=The index 'i' in array_write must be Variable in dygraph moder/r0rr zNThe index 'i' should not be greater than the length of 'array' in dygraph moder1r array_writez'x should be pir.OpResult, but recevied .r3xz7array should be tensor array vairable in array_write Op.outnamerrwrite_to_arrayr5rr)r;)r rrr7r8 create_arrayrrrappendrrrrrr rrr! array_write_rrr%r r"r#r$create_variabler@r()r=r1r)r*s r,r;r;s@ x  P O P  x  K J K ww    B A B  FF1I = )E 4  G F G C    \ [ \  s5z>E!H  LLO  C'MB!VZZ0019$q'!D   ufjj&9&9:779 GHH =OO009E,,UAq9  C'MB1cH}57fh7  uh/::!5!5!F!FFM =** }D)\\))::gg+E !A3'UG$   r-cg}|Dt|ttfs#tdj t |t|}|D]O}t|t tjjfr.tdj t |tr|Strt|tjjtjfs)tj j"j%|}tj&j)|}|D],}tj&j+||t-|}.|St/dit1}|j3|j4dtjjj6|}|D]}t9|t-|||S)aR This OP creates an array. It is used as the input of :ref:`api_paddle_tensor_array_array_read` and :ref:`api_paddle_tensor_array_array_write`. Args: dtype (str): The data type of the elements in the array. Support data type: float32, float64, int32, int64 and bool. initialized_list(list): Used to initialize as default value for created array. All values in initialized list should be a Tensor. Returns: list|Tensor: An empty array. In dynamic mode, ``array`` is a Python list. But in static graph mode, array is a Tensor whose ``VarType`` is ``LOD_TENSOR_ARRAY``. Examples: .. code-block:: python >>> import paddle >>> arr = paddle.tensor.create_array(dtype="float32") >>> x = paddle.full(shape=[1, 3], fill_value=5, dtype="float32") >>> i = paddle.zeros(shape=[1], dtype="int32") >>> arr = paddle.tensor.array_write(x, i, array=arr) >>> item = paddle.tensor.array_read(arr, i) >>> print(item.numpy()) [[5. 5. 5.]] zDRequire type(initialized_list) should be list/tuple, but received {}zUAll values in `initialized_list` should be Variable or pir.OpResult, but recevied {}.r>r?)r=r1r))r))rrtupler formatrrrrrr rr r"r#DataTypebase frameworkconvert_np_dtype_to_dtype_r!rBrDrrr%rEr@r$r;)rinitialized_listr)valr9r* tensor_arrays r,rBrBs< E#*T5M:V]])*  %&#&***=*=>?gnnI  %$,,"6"6 !FGKK))DDUKEoo**51C//..sCc9JKC 11--KK=%%%66. C #l!;< Pr-)N)rbase.data_feederrrbase.frameworkrcommon_ops_importrrKrr r __all__rr2r;rBr-r,rUs8"C((:: 8vK\`FEr-