j ^ddlmZddlmZmZmZmZmZddlm Z ddl m Z ddl m Z edZeeedfeefZed d ed ed eeeeffd Zdeed ed eeeeffdZded efdZdedeed ed eefdZdS)) lru_cache)ListSequenceTupleTypeVarUnion) EinopsError) get_backend)ParsedExpressionTensor.)maxsizepatternopnamereturnc |}t|}t|t|krtd|d|dd|vrtd|d|d|D]:}|dkr2t j|\}}|std|d|d|d;|d}t||z dz }||z} ||| fS) NzDuplicates in axes names in z(..., "z")*z No *-axis in zInvalid axis name z in )splitsetlenr r check_axis_name_return_reasonindex) rraxesaxes_setaxisis_validreason n_axes_before n_axes_aftermin_axess ?C:\Projects\Historical\venv\Lib\site-packages\einops/packing.pyanalyze_patternr# s! ==??D4yyH 4yyCMM!!SSSSSSTTT (D&DDDDDEEE]] 3;;/MdSS Hf ]!"[t"["["["[PW"["["[\\\JJsOOMt99},q0L|+H , 00tensorsc t|d\}}}t|d}g}g}t|D]\}} || } t | |krt d|d| d|d|d t | |z } || || ||| g| d|d | | dR||| |fS) a Packs several tensors into one. See einops tutorial for introduction into packing (and how it replaces stack and concatenation). Parameters: tensors: tensors to be packed, can be of different dimensionality pattern: pattern that is shared for all inputs and output, e.g. "i j * k" or "batch seq *" Returns: (packed_tensor, packed_shapes aka PS) Example: ```python >>> from numpy import zeros as Z >>> inputs = [Z([2, 3, 5]), Z([2, 3, 7, 5]), Z([2, 3, 7, 9, 5])] >>> packed, ps = pack(inputs, 'i j * k') >>> packed.shape, ps ((2, 3, 71, 5), [(), (7,), (7, 9)]) ``` In this example, axes were matched to: i=2, j=3, k=5 based on order (first, second, and last). All other axes were 'packed' and concatenated. PS (packed shapes) contains information about axes that were matched to '*' in every input. Resulting tensor has as many elements as all inputs in total. Packing can be reversed with unpack, which additionally needs PS (packed shapes) to reconstruct order. ```python >>> inputs_unpacked = unpack(packed, ps, 'i j * k') >>> [x.shape for x in inputs_unpacked] [(2, 3, 5), (2, 3, 7, 5), (2, 3, 7, 9, 5)] ``` Read the tutorial for introduction and application scenarios. packrzpacked tensor #z' (enumeration starts with 0) has shape z, while pattern z assumes at least z axesN)r) r#r enumerateshaperr appendreshapeconcat) r%rrr r!backendreshaped_tensors packed_shapesitensorr*axis_after_packed_axess r"r'r'!smH->*> ? ? NNr$xcd}|D]}||z}|S)Nr)r4resultr1s r"prodr8Zs& F !  Mr$r2r0c t|d\}}}t}t||dz|zkrt d|d||}d|D}t d|D} | dkrt d|d|d d gt|z||gz| d kr/t |d d D]\} } | | z| dz<n|d } t| D]} | || z| dz<t| dzt|d d d D]} | dz|| z | <|d |||dzd td d f|z fd t |DS#t$r$}t d|dd d||d }~wwxYw)a8 Unpacks a single tensor into several by splitting over a selected axes. See einops tutorial for introduction into packing (and how it replaces stack and concatenation). Parameters: tensor: tensor to be unpacked packed_shapes: packed_shapes (aka PS) is a list of shapes that take place of '*' in each output. output will contain a single tensor for every provided shape pattern: pattern that is shared for input and all outputs, e.g. "i j * k" or "batch seq *", where * designates an axis to be unpacked Returns: list of tensors If framework supports views, results are views to the original tensor. Example: ```python >>> from numpy import zeros as Z >>> inputs = [Z([2, 3, 5]), Z([2, 3, 7, 5]), Z([2, 3, 7, 9, 5])] >>> packed, ps = pack(inputs, 'i j * k') >>> packed.shape, ps ((2, 3, 71, 5), [(), (7,), (7, 9)]) ``` In this example, axes were matched to: i=2, j=3, k=5 based on order (first, second, and last). All other axes were 'packed' and concatenated. PS (packed shapes) contains information about axes that were matched to '*' in every input. Resulting tensor has as many elements as all inputs in total. Packing can be reversed with unpack, which additionally needs PS (packed shapes) to reconstruct order. ```python >>> inputs_unpacked = unpack(packed, ps, 'i j * k') >>> [x.shape for x in inputs_unpacked] [(2, 3, 5), (2, 3, 7, 5), (2, 3, 7, 9, 5)] ``` Read the tutorial for introduction and application scenarios. unpack)rrz unpack(..., z)) received input of wrong dim with shape c8g|]}d|vrdnt|S)r()r8).0p_shapes r" zunpack..s+*o*o*oV]w22DMM*o*o*or$c3<K|]}t|dkVdS)r(N)int)r<r4s r" zunpack..s,!Q!Q1#a2g,,!Q!Q!Q!Q!Q!Qr$z) received more than one -1 in z and can't infer dimensionsrNr(c g|]J\}}gt||dzRg|RKS)r)r,slice) r<r1 element_shaper. shape_end shape_start slice_fillersplit_positionsr2s r"r>zunpack..s   != OOYYu_Q-?QRUVQVAW'X'XYYZ:+: : ::     r$zError during unpack(..., "z!"): could not split axis of size z into requested ) r#r r*rr sumr)rrangerC Exception)r2r0rrr r! input_shape unpacked_axislengths_of_composed_axesn_unknown_composed_axesr1r4unknown_composed_axisjer.rErFrGrHs` @@@@@r"r:r:asR-%C%CB%G%G,-- V VA%4Q%7:RST:U%UOAE " ",q0#6N2O2OPPQUQUSUQUV V VA!0Q!7:RST:U!UOA  n}n-KMA-//0I$%%'-7L         %.m$<$<       / / /SbceSf / /, / /   s0G G>G99G>N) functoolsrtypingrrrrreinopsr einops._backendsr einops.parsingr r r@Shapestrr#r'r8r:r6r$r"rZs88888888888888((((((++++++    eCHotCy() 31S1#1%S# 2F1111&6O(6"6OS6OU64;;N5O6O6O6O6OrEc\6\$u+\\V \\\\\\r$