\;jKXddlZddlmZddlmZddlmZddlmZddl Z ddl Z ddl m Z ddlmZmZdd lmZmZmZmZefd Zd ZGd d eZeZdZdZdZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(d(dZ)d)dZ*dZ+dZ,d*d Z-d!Z.d+d"Z/d#Z0d$Z1d%Z2d&Z3d'Z4y),N) defaultdict)Sequence)uuid4)WeakKeyDictionary)convert_np_dtype_to_dtype_) check_dtype convert_dtype)BlockVariable_current_expected_placein_dygraph_modec t||r|g|zS t|}t ||k7r,td|zdzt |zdzt |z|D]?}t|t tjjfr Jd|z ||A|S#t$rtd|zdzt |zwxYw#ttf$rftd|zdzt |zdzt |zdzt |zd zt |zd zdzt t|zwxYw) am Converts a single numerical type or iterable of numerical types into a numerical type list. Arguments: value: The value to validate and convert. Could an int, or any iterable of ints. n: The size of the list to be returned. name: The name of the argument being validated, e.g. "stride" or "filter_size". This is only used to format error messages. dtype: the numerical type of the element of the list to be returned. Returns: A list of n dtypes. Raises: ValueError: If something else than an int/long or iterable thereof was passed. zThe z)'s type must be list or tuple. Received: z's length must be z . Received: z7Required numerical type with '%s', but received Tensor.z#'s type must be a list or tuple of  z . Received: z including element z of type) isinstancelist TypeError ValueErrorstrlenr paddlepirOpResulttype)valuennamedtype value_list single_values bG:\00. PROJECTS\API\Inventory\templateJSON\kerjaOCR\Lib\site-packages\paddle/utils/layers_utils.pyconvert_to_listr"#s(%    eJ z?a &'a&! ! e*  'L!x)<)<= J   l#'4U =>e*  2 *  ;<!f  %j ! & &%j!) ),' (! ! $|,- . s BC 'C A5D>cjt|tryt|txrt|t S)z7 Whether `seq` is an entry or nested structure T)rdictrr)seqs r! is_sequencer&ks-#t c8 $ AZS-A)AAceZdZfdZxZS) UniqueIdMapcLt||tt|_yN)super__init__rrdata)self __class__s r!r-zUniqueIdMap.__init__us & r')__name__ __module__ __qualname__r- __classcell__)r0s@r!r)r)ts ''r'r)ct|tr t|fSt|tr t |fSt |j fSr+)rrhashrid uniqueidmapint)objs r!uniqueidr;}sB#sS | C 3zC $$&&r'cft|dkDsJd}|D]}|t|z}t|S)zM Return int hash value calculated by id(arg) or tuple(id1,id2, ...). r)rr;r6)argsinfovs r! _hash_with_idrAs; t9q== D hqk! :r'cf t|jS#t$r tdwxYw)zR Returns a sorted list of the dict keys, with error if keys not sortable. z,nest only supports dicts with sortable keys.)sortedkeysr)dict_s r!_sortedrFs6Hejjl## HFGGHs0c#tKt|trt|D] }|| y|Ed{y7wr+)rr$rF)iterablekeys r! _yield_valuerJs6(D! 8$C3- %s .868c#vKt|D]%}t|rt|Ed{"|'y7 wr+)rJr&_yield_flat_nest)nestrs r!rLrLs2 $  q>'* * *G  *s (97 9c"t|r|S|gSr+r&rMs r! to_sequencerQs4 v r'cFt|rtt|S|gS)z :alias_main: paddle.flatten :alias: paddle.flatten,paddle.tensor.flatten,paddle.tensor.manipulation.flatten :old_api: paddle.base.layers.flatten Traverse all entries in the nested structure and put them into an list. )r&rrLrPs r!flattenrSs$4$T*++v r'ct|trFttt||t |fd|j DSt|t rPt|drDt|jtr*td|jDrt ||St ||S)zE Convert the sequence `args` to the same type as `instance`. c3,K|] }||f ywr+r=).0rIresults r! z!_sequence_like..sLOSsF3K0Os_fieldsc3<K|]}t|tywr+)rr)rVfs r!rXz!_sequence_like..s=,rWs @r!_sequence_likeras(D! c'(+T23tH~LHMMOLLL8U# Hi ( x'' 2 =H,<,<= =tH~t$$tH~d##r'cg}t|D]T}t|r.t|||\}}|jt |||}<|j|||dz }V||fS)z/ Helper function for pack_sequence_as. )rJr&_packed_nest_with_indicesappendra) structureflatindexpackeds new_indexchilds r!rdrdsoF ) $ q>8D%H Iu MM.E2 3E MM$u+ & QJE% &=r'c\t|s tdt|s*t|dk7rtdt|z|dSt |}t|t|k7r$tdt|t|||fzt ||d\}}t ||S)zA Pack a given flattened sequence into a given structure. z flat_sequence must be a sequencercz6Structure is a scalar but len(flat_sequence) == %d > 1rzyCould not pack sequence. Structure had %d elements, but flat_sequence had %d elements. Structure: %s, flat_sequence: %s.)r&rrrrSrdra)rf flat_sequenceflat_structure_ris r!pack_sequence_asrqs } %:;; y ! }  "Hm$% QY'N >c-00 BN#M"    *)]AFIAv )V ,,r'c |Dcgc] }t|}}t|}t|d|Dcgc]}|| c}Scc}wcc}w)zO Apply `func` to each entry in `structure` and return a new structure. r)rSr\rq)funcrfrjroentriesxs r! map_structurerv sR+44)Qgaj)N4>"G IaLW*EW48W*E FF5*Es A A c,|D]}t|syy)zD Returns whether structure holds sequence like `list/dict`. TFrO)rfrjs r!hold_mutable_varsrxs q> r'cVtjt|}t||S)zE Returns vars copied from sequence without mutable property. )copyrSrq)rfros r!copy_mutable_varsr{s$YYwy12N I~ 66r'c&t|}|t|k7rtd|d|d|sy|rt|}t|}||k7rtdj ||t |t rRt|j}t|j}||k7rtdj ||tt|}tt|} t|| D]\} } t| | |y)z6 Helper function for `assert_same_structure`. zKThe two structures don't have the same nested structure. First structure: z Second structure: .NzvThe two structures don't have the same sequence type. First structure has type {}, while second structure has type {}.zvThe two dictionaries don't have the same set of keys. First structure has keys {}, while second structure has keys {}.) r&rrrformatrr$setrDrrJr\ _recursive_assert_same_structure) nest1nest2 check_typesis_sequence_nest1 type_nest1 type_nest2keys1keys2nest1_as_sequencenest2_as_sequencen1n2s r!rr&s&$E*K.. %w&._padding_to_same_structure_single..change_none_to_objNsy Hr')r&rqrS)rr:ritems ` r!!_padding_to_same_structure_singlezDpadding_to_same_structure.._padding_to_same_structure_singleMsX  u $WU^L^T*40^LE  'u-E MsA r=)rrr:rs r!padding_to_same_structurerLs+  .eS 9E -eS 9E %<r'ct|rtt|nd}t|rtt|nd}||k7rtd||||fzt |||y)z@ Confirm two nested structures with the same structure. rczThe two structures don't have the same number of elements. First structure (%i elements): %s Second structure (%i elements): %sN)r&rrSrr)rrr len_nest1 len_nest2s r!assert_same_structurer`sm(35'9GEN#qI'25'9GEN#qII 1%E2 3  %UE;?r'ct||dzk(st||k(sJd}t||dzk(r't|D]}||dz||dzdzk7sd}|S)z/ Check whether padding is symmetrical. rTrcF)rrange)paddingdata_dimis_sysis r!_is_symmetric_paddingrpsq w<8a< '3w<8+CC C F 7|x!|#xAq1u~Q!33! Mr'cj|D].}t|ttjjfs.yy)zC Check whether list or tuple contains variable / OpResult. TF)rr rrr) list_or_tuplers r! _contain_varr}s/ dXvzz':':; < r'c | t}g}|D]}t|tjjrmd|_t |j|k7rtj||}|jgk7rtj|g}|j|tjg|ttj||}|j||S)NTrur)r rrrr stop_gradientr rcastshapereshaperefullrnp)ele_listplace default_dtypeint_tensor_listeletemp_outs r!get_int_tensor_listrs }')O c6::.. / $C SYY'=8kkC}=yyBnnS"-  " "3 '{{*288M+BC H  " "8 , r'cddlmd}fd}t|tr\d|_t |j dddgd d zd zt|j dk(rtj|d}||d <yt|ttfr#|||d<t|r |||d <yytd)Nr fill_constantcg}t|D]8\}}t|tr|jd(|j|:|S)N) enumeraterr re) list_shape attr_shapeidxdims r!_get_attr_shapez0get_shape_tensor_inputs.._get_attr_shapesH !*-HC#x(!!"%!!#& . r'c |g}t|D]\}}t|trxd|_t |j dt |zdzddgdzdzt|j dk(rtj|d}|j|gd|d }|j||S) NTzshape[]int32int64(When type of shape inzis list or tuple.)r force_cpu) rrr rr rrr rrre)rshape_tensor_listrrrrop_types r!_get_shape_tensorz2get_shape_tensor_inputs.._get_shape_tensors!*-HC#x($(!IIs3x'#-g&,w69MM !+w6 ++7;C!((-(WcTJ!((2. ! r'Trrrrrz is Variable.) ShapeTensorShapeTensorListz0Shape only supports Variable, or list, or tuple.) paddle.tensorrrr rr rr rrrr]rr)inputsattrsrrrrrs ` @r!get_shape_tensor_inputsrs+!(%"" KK  g   $w .1A A    % 0KKw/E %} ED%= )(/g  (9%(@F$ % JKKr'c ddlm}g}|D]u}t|ttj j frd|_|j|Ft|tsJ|dg||d}|j|w|S)zA Converts all elements of a list to Variable / OpResult. rrTrcr) rrrr rrrrrer9)old_listrrnew_list_tensorrrs r!_convert_to_tensor_listrs},O cHfjj&9&9: ; $C   " "3 'c3' ''$aS%EH  " "8 , r'ct|ttfr2|Dcgc]%}t|tr|j dn|'}}|St r#|j tj}|Scc}w)zI Convert shape(list, tuple, variable) to list in imperative mode r) rrr]r rrastyper9tolist)rrus r!convert_shape_to_listrsm%$'FKLejH51}t|tr|dkr t dt|t r5t dy) zU Check shape type and shape elements type before passing it to fill_constant rrrrrzDAll elements in ``shape`` must be positive when it's a list or tuplezDAll elements in ``shape`` must be integers when it's a list or tupleN)rr r rrr9r)rrs r! check_shapersl%"EKK7G*rs #$% 69+.EPB'#' m 'H  $2  -:G7#>L( @ 01Lh$ &1.:'r'