o !j:v@sddlZddlZddlZddlmZmZdZddZddZ d d Z d d Z d dZ ddZ ddZddZ d"ddZddZddZddZddZd d!ZdS)#N)core unique_nameicsddlmt|}fdd|D}|t}|dkr|S|dkr&td|t}|t|dkr9|ddStdgt|j t||dd|||d<|S)NrVariablecs.g|]}t|tjjtjfs|dur|qSN) isinstancepaddlepirOpResultnpndarray).0eler[/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/base/variable_index.py (sz$replace_ellipsis..rz0An index can only have a single ellipsis ('...')) frameworkrlistcountEllipsis IndexErrorindexlensliceshape)varitemZitem_remove_varZ ell_countZell_idxrrrreplace_ellipsiss"     rcCsTg}|D]#}t|tjr|t|qt|tr"|t|q||q|Sr)rr r appendr assignranger)rnew_item slice_itemrrrreplace_ndarray_and_range@s   r%cCs@g}g}t|D]\}}|dur||q||q||fSr) enumerater )rr# none_axesir$rrr replace_noneLs  r)cCsddlm}t|turdSt||r@tddr/t|jdkr/|jddkr/t ddSt|jdkr>|j tj kr>dSdSt|tj jrXt|jdkrX|j tjjjkrXdSdS)NrrTZFLAGS_set_to_1drz1-D Tensor will be treat as advanced indexing in future version. Currently, 1-D Tensor means a scalar, not vector, and please modify it to 0-D Tensor. If advanced indexing is needed, please use `export FLAGS_set_to_1d=False` to set the flag.F)rrtypeintrr Z get_flagsrrwarningswarndtypeboolr r base libpaddleBOOLrrrrris_integer_or_scalar_tensorWs"   r4cCsLddlm}t||r|jtjkrdSt|tjjr$|jtjj j kr$dSdS)NrrTF) rrrr.r r/r r r0r1r2r3rrris_bool_tensorns  r5c Csddlm}tj|r=tjj|dd||<t|D] \}}t||tjj fr3|| dd||<q|| |qdS|||<dS)Nrrint64r.r) rrr utils _contain_var_convert_to_tensor_listr&rr r r ) attrsattr attr_nameZtensor_attr_nameinputs infer_flagsrr(dimrrr deal_attrs{s     rAcCst|jt|jkrtdt|jdt|jdd}|j}|t|jkrT||}|dkrI|j|dkrI||j|krItd||j|||d7}|t|jks't|}t||S)NzkThe dims of bool index doesn't match indexed array, the dims of bool index except to be equal or less than z, but received .rrzxThe dimension of bool index doesn't match indexed array along dimension {}, the target dimension is {}, but received {}.r)rrrformatr nonzero gather_nd)rrr(Z item_shapeZdim_lenZ bool_2_idxrrrget_value_for_bool_tensors.$  rFcCs|ddlm}trJdt||tfr5ddlm}ddlm }tj ||dd}||}||||d St d t |) abranches for tensor array setitem operation. A item can be a: (1) int/Variable, which is a simple number/variable such as [1], [-2] (2) Slice, which is represented by bounds such as [2:-1] (3) Tuple, which includes the above two cases such as [2:-1, 1] If item is case (1), we perform paddle.tensor.array_write, in other cases, we raise a NotImplementedError. rrz=setitem for tensor_array must be called in static graph mode.r)to_static_variable) array_writer6r7)xr(arrayzEOnly support __setitem__ by Int/Variable in tensor_array, but gets {})rrr in_dynamic_moderr+Z(paddle.jit.dy2static.variable_trans_funcrGZ paddle.tensorrHcastNotImplementedErrorrCr*)rrvaluerrGrHrrr_setitem_for_tensor_arrays   rOTcCs8g}g}t}d}t|D]<\} } | durH| dkrd}| dkr/t|dkr/|d| dkr/d}nt|| }t|| dj}|| || dq t|jD] } || dur[|| qN|ret | ng} d} |t t|jkry|} |rx|} nd}| |} |r|jdkr|dkr| |} n|} | || ||| |fS)an Transpose origin Tensor and advanced indices to the front. Returns: transed_tensor (Tensor): transposed tensor, corresbonding with advanced indices transed_index (List): advanced indices transed to the front trans_back_dim (List): order of axes to transpose back to original order. Only used in __setitem__. pos_of_new_dim (int): axis of new dim in the result. Only used in __getitem__. rank_of_new_dim (int): rank of new dim in the result. Only used in __getitem__. transed_value_tensor (Tensor): value tensor transed to the front. Only used in __setitem__. rNrrT) MAX_INTEGERr&rminmaxndimr r"r Zargsorttolistr transpose)Z ori_tensorindicesZis_for_setitemvalues out_is_viewZ transed_dimZ transed_indexpos_of_new_dimrank_of_new_dimr(indiceZtrans_back_dimZtransed_value_tensortransed_tensorrrrdeal_advanced_indexsP$      r]c Cs^ddlm}|r|}nt|do|jtjjj k}|r!gn dgdt |j }g}g}g}g}g} d} d} t |t sA|f}t|}t||}t|\}} d} d}t|D]\}}d\}}}t|r|st |tr|j |dur|j |dkr||j |krtd||||j |f|||}d}|d kr|dnt}|d7}nAt |tr| || t|gf|| <d } | d7} n%t |tr |j}|j}|j}| d7} |d7}|dur|dur|durqX|durdn|}|dur|dkrdnt}|dur |dkrtnd }nt |tt frJ| t|f|| <|| djtjkr?t ||j |kr?td t ||j ||d } | d7} |d7}nt |tj!j"r|jtjksb|jtj!j#j$kr|j%dkrn| |n|j d|j |krtd |j d|j ||| |f|| <d } | d7} |d7}nPt |tj&j'r|jtj&jj(j$kr|j%dkr| |n|j d|j |krtd |j d|j ||| |f|| <d } | d7} |d7}ntd ||dus#|dus#|dus#||||| |||dt |tj!j"tj&j'fs|dkr!d n| } qX||| || ||| | f S) Nr) in_pir_modedescFr)NNNz= 0 and < x.shape[%d]: %drTzKThe shape of boolean index {} did not match indexed tensor {} along axis {}zdValid index accept int / bool / slice / ellipsis / list / Tuple / Ndarray / Tensor, but received {}.))rr^is_dense_tensor_array_typehasattrr_r*rVarDescVarTypeLOD_TENSOR_ARRAYrrrtupler%rr)r&r4r+rr rPr/r Z to_tensorrstartstopsteprr.rCr0rr1r2rSr r ZDataType)rIrVr^Zis_tensor_arrayadvanced_index decrease_axesaxesstartsendsstepsuse_strided_slicehas_advanced_indexr'Z estimated_dimr@r(r$rgendrirrr parse_indexs                             rsc" Cstddlm}ddlm}m}|jtjjj j j krt |||St ||\ }}}} } } } } }d|i}| |||| | d}d}d}d}d}d}tj|rVtj|}||d<|d=tj|ritj|}||d <|d =tj|r|tj|}||d <|d =| sB|j}||d <ddlm}t|ttttfrt|g||}t|tjrt|j}| }||d<||d<nt||tj!j"fr||}||d<|}n t#dt||r|durtj$%||||| | | || Stj$&|||||| | | Stjj'j( dit)}|j*j+dkr|j,|jd}n|j-|jd}|.}|j/d|d|i|ddidtj0j1j23j45|j6|j78||St||tj!j"tj!j9fsXt:||j}t;|| |||| | |\}}t<|| d||\}}}} } }}|j|jkr||j}t=rt>|dkr|djtjtjj?j@fvrt>|djt>|jkr|j|jkr|A|j}tj$BtC|d||}|s|Sn|D||}|s|Sn|E||}|F|}!|!|d<|r|G|}n"tjj'j( dit)}|j*j+dkr |j,|jd}n|j-|jd}|.}|j/d|d|i|ddid|s8tj0j1j23j45|j6|j78||S)a In dynamic mode, this function will modify the value at input tensor, returning same Tensor as input. But it will return a new Tensor with assigned value in static mode. Args: x(Tensor): Tensor to be set value. indices(int|slice|None|Tensor|List|Tuple...): Indices, used to indicate the position of the element to be fetched. values(Tensor|Number|Ndarray): values to be assigned to the x. r)in_dynamic_or_pir_mode)rdefault_main_programInput)rlrmrnrorkr'NStartsTensorListrmEndsTensorListrnStepsTensorListror.) convert_dtyperWrZ ValueTensorzjOnly support to assign an integer, float, numpy.ndarray or paddle.Tensor to a paddle.Tensor, but received set_valuerr7Out)r*r>outputsr;Z inplace_mapT)r{)Hrtrrrur*r r0rrcrdrerOrsr8r9r:r.Z data_feederrzrr/r+floatcomplexr rJZastyper rrZravelrTr Value TypeError_C_opsZ set_value_Zset_value_with_tensor_Z layer_helperZ LayerHelperlocalsZ main_programZcurrent_block_idxZ*_create_global_variable_for_type_inferenceZ"create_variable_for_type_inference current_block append_opZjitapiZProgramTranslatorZ get_instanceZ _inplace_mapaddprogramr_idr r!get_tensor_with_basic_indexingr]rKrr1r2expandZwhere_Z logical_notZ index_put_Z index_putrUZ_bump_inplace_version)"rIrVrWrtrrurmrnrorlr'rkrjrqrpr>r;Z value_tensorrwrxryrr.rzhelperoutputZ cur_blockZ sub_tensorZis_viewZtransed_sub_tensoradjusted_advanced_indexZ transback_dim_Ztransback_sub_tensorrrr_setitem_statics                           rc sddlm}d} |rt|drd|_t|dkr|} nd} |r#dnd} d |gi} |gg|d } |r7g| d <dgt|}t| |d d | |t| |dd| |t| |d d| ||| d<ddlm}m}|rd | vrs| d }n| d }d| vr| d}n| d}d| vr| d}n| d }|rt j |||||} t|dkrt j | |} no|rt|ttfrt j|rt j|}t|ttfrt j|rt j|}|rt j||dfSt j ||||| d| d} n)ddlm}|}|jt|jd| |jd}|j | | d|gi| d|} t|dkrKd} t!|D]\}tfdd|D}|}|||<q+t j"| |d} |rXt| drXd| _| | fS)Nr)in_to_static_modeF is_view_varTr strided_slicerrv)rlrmrn decrease_axisstridesrmrwrnrxZStridesTensorListr?)rtr^r)rur)namer.r|)r*r>r}r;csg|]}|kr|qSrr)rr(Zaxisrrrsz2get_tensor_with_basic_indexing..r)#Z dygraph.baserrbrrrAr~rtr^keysr rrZsqueezerrrfr8r9Zget_int_tensor_listraZ_pir_opsZslice_array_denserrrurZ create_varrZgenerate_with_ignorable_keyrr.rr& unsqueeze)rIrlrmrnrorkr'rprrXoutZop_typer>r;r?rtr^strrZstrideruZ target_blockZ slice_out_varidxlZnew_axisrrrrs                    rc Cst||\ }}}}}}}} } t|||||||| \} } | rt| |dd\} }} }}} } t|dkrF|djtjtjjj fvrFt | |d} nt |}t|dkrXtj |dd}n|d d}t| |} |dkrtt|||ttd|tt||| j}| |} | S)z Args: x(Tensor): Tensor to be indexing. indices(int|slice|None|Tensor|List|Tuple...): Indices, used to indicate the position of the element to be fetched. FNrrrr)rsrr]rr.r r/r0r1r2rF parse_bool_and_broadcast_indicesstackrrErr"rSrU)rIrVrmrnrorlr'rkrjrqrprrr\rrYrZZadvanced_index_tensorpermrrr_getitem_staticsr     rcCsRt|D]\}}|jtjkrt|dddf||<qt|dkr't|}|S)Nrr)r&r.r r/rDrZbroadcast_tensors)rVr(r[rrrr\s   r)T)r,numpyr r r~rrrPrr%r)r4r5rArFrOr]rsrrrrrrrrs,%    D-sk M