&VfjjddlmZdZdZddZdZdZdZdd Zdd Z dd Z d Z dZ dZ dZd S))dmtreec|SN)clss W/var/www/html/software/conda/lib/python3.11/site-packages/keras/src/tree/dmtree_impl.pyregister_tree_node_classr s Jc.tj|dSr)r is_nested structures rr r s Yr Tc0tj|||S)N)top_down)rtraverse)funcrrs rrr s ?4X > > >>r c*tj|Sr)rflattenr s rrrs >) $ $$r c&tj|g|RSr)r map_structure)r structuress rrrs   2z 2 2 22r c(tj||g|RSr)rmap_structure_up_to)shallow_structurerrs rrrs  %&7 K K K KKr c0tj|||S)N) check_types)rassert_same_structure)abrs rrrs  '1+ F F FFr Nc Dtj}|p tj}d}||s:td||dt |||s|t |dkratdt |||dt |t |||d|dS t||d||\}}|t |krtnz#t$rmtj |}t |t |kr6tdt |dt |d |d |d YnwxYw|||S) NcNt|}|d|||dodzS)Nz...)str)valuelength value_strs rtruncatez"pack_sequence_as..truncate$s/JJ &!Yvww%7%AEBBr zYAttempted to pack value: {} into a structure, but found incompatible type `{}` instead.dzThe target structure is of type `{}` {} However the input is a sequence ({}) of length {}. {} nest cannot guarantee that it is safe to map one to the other.rz'Could not pack sequence. Structure had z atoms, but flat_sequence had z items. Structure: z, flat_sequence: .) rr _sequence_like TypeErrorformattypelen ValueErrorpacked_nest_with_indices IndexErrorr)r flat_sequence sequence_fn is_nested_fnr& final_indexpackedflat_structures rpack_sequence_asr8 s*#L6!6KCCC < & &  ..4f,,d=.A.A//    < " "  }   " "EEKVOOHY,,'' &&H]C00 FF   Q6 }a{   V ]++ + +  ,     22 ~  #m"4"4 4 4K!$^!4!4KK%(%7%7KK(KK;HKKK  5 4  ;y& ) ))s+0DA4FFcg}|p tj}t|D]e}||r8t|||||\}}|||||}E||||dz }f||fS)aHelper function for pack_sequence_as. Args: structure: structure to mimic. flat: Flattened values to output substructure for. index: Index at which to start reading from flat. is_nested_fn: Function used to test if a value should be treated as a nested structure. sequence_fn: Function used to generate a new structure instance. Returns: The tuple (new_index, child), where: * new_index - the updated index into `flat` having processed `structure`. * packed - the subset of `flat` corresponding to `structure`, having started at `index`, and packed into the same nested format. r()rr* yield_valuer0append) rflatindexr4r3r6s new_indexchilds rr0r0Rs*F6!6K  # #   <?? 74 k   Iu MM++a// 0 0 0EE MM$u+ & & & QJEE &=r c#FKtj|D] \}}|V dSr)r_yield_sorted_items)iterable_vs rr:r:vs9*8441r cPd}t|tj||S)Nctt|trt|Stj||Sr) isinstancelisttuplerr*)instanceargss rr3z$lists_to_tuples..sequence_fn|s3 h % % ;; $Xt444r )r3)r8rr)rr3s rlists_to_tuplesrM{s=555 y!!   r ctt|ttfrtd|DrdSdS)Nc3^K|](}t|ttdfV)dSr)rHintr-).0es r z!is_shape_tuple..s7;;Az!c4::.//;;;;;;r TF)rHrIrJall)xs ris_shape_tuplerVsA!dE]## ;;;;; ; ; 4 5r ct|rt|St|trfd|DSt|trtfd|DSt|tr fd|DSt d|)Nc0g|]}t|Srmap_shape_structurerQrRrs r z'map_shape_structure..s$@@@#D!,,@@@r c38K|]}t|VdSrrYr[s rrSz&map_shape_structure..s.EEa(q11EEEEEEr c8i|]\}}|t|SrrY)rQkrErs r z'map_shape_structure..s*NNNDAq&tQ//NNNr z&Cannot map function to unknown object )rVrJrHrIdictitemsr/)rrs` rrZrZsi  &tE)$$%%%)T""A@@@@i@@@@)U##FEEEE9EEEEEE)T""ONNNNIOOrds //////   ????%%%333LLLGGGG/*/*/*/*f7;!!!!H    O O O O Or