jUddlZddlZddlmZmZmZmZmZddlm Z dZ e e d<GddZ Gdd ZdS) N)ListOptionalSetTupleUnion) EinopsErroru… _ellipsisc$eZdZdZdefdZdZdS) AnonymousAxiszHImportant thing: all instances of this class are not equal to each othervaluect||_|jdkr1|jdkrtdtd|jdS)NzENo need to create anonymous axis of length 1. Report this as an issuez0Anonymous axis should have positive length, not )intr r)selfr s ?C:\Projects\Historical\venv\Lib\site-packages\einops/parsing.py__init__zAnonymousAxis.__init__ sVZZ :??zQ!"ijjj!"aUYU_"a"abbb ?c0t|jdS)Nz-axis)strr )rs r__repr__zAnonymousAxis.__repr__sdj//((((rN)__name__ __module__ __qualname____doc__rrrrrr r sIRRcccccc)))))rr c eZdZdZddddededefdZdefd Zdefd Z e dd edede eeffd Z e d edefd Z dS)ParsedExpressionz non-mutable structure that contains information about one side of expression (e.g. 'b c (h w)') and keeps some information important for downstream F)allow_underscoreallow_duplicates expressionrrc(d_d_t_d_g_d|vrd|vrt dt|ddkst|ddkrt d| dt}d_dfd }d}|D]}|d vr]| ||d}|d krt d g/|d kr-t dj dct |s |dtfvr ||}||z }t d|dt d|d| ||dSdS)NF.z...z6Expression may contain dots only inside ellipsis (...)rzUExpression may contain dots only inside ellipsis (...); only one ellipsis for tensor Tc|jvrr|dksstd|d|tkrljt(jtd_dStd_dSt|}|r3t|dkr jgn dS |\}}|s|std|d ||rt|}j||rd_ j|gdS|dS) N_z2Indexing expression contains duplicate dimension ""FTr)rzInvalid axis identifier:  ) identifiersrr add compositionappendhas_ellipsis_parenthesizedr isdecimalrcheck_axis_name_return_reasonr has_non_unitary_anonymous_axes)x is_number is_axis_namereasonrr bracket_grouprs r add_axis_namez0ParsedExpression.__init__..add_axis_name3sD$$$(aQ#XX?OX%&_[\&_&_&_```I~~ $$Y/// ($++I6666;D333!((3336:D333MM!,, Q1$,(//3333F'+'I'I!^n'I'o'o$ f!Q\Q%&O!&O&Ov&O&OPPP)%a((A $$Q'''?:>D7 ($++QC00000!((+++++rz() (zDAxis composition is one-level (brackets inside brackets not allowed))zBrackets are not balancedr%zUnknown character ''z'Imbalanced parentheses in expression: "r&) has_ellipsisr,setr(r/r*rrcountreplacer r+isalnum)rr rrr5current_identifiercharr4s` `` @rrzParsedExpression.__init__s*"':>'%(UU49+8: *  J&&!"Z[[[yyU++q00CIIj#4N4NRS4S4S!k$++E9==J $D -1  , , , , , , , ,D" A ADu}}%1!M"4555%)"3;;$0)*pqqq$&MMS[[$,)*EFFF$++M:::$(MT"" AdsI.>&>&>%-)-&&&$.&&!"?"?"?"?@@@  $U UUUVV V  ) M, - - - - - * )rreturncg}|jD];}t|ts Jd|D]}||<|S)Nzdoes not work with ellipsis)r* isinstancelistr+)rresult composed_axisaxiss rflat_axes_orderz ParsedExpression.flat_axes_orderqsf!- $ $MmT22 Q Q4Q Q Q2% $ $ d#### $ rcp|jD]-}t|trt|dkrdS.dS)NrTF)r*rBrClen)raxess rhas_composed_axesz"ParsedExpression.has_composed_axesysA$  D$%% #d))a--tturnamec,t|sdS|ddks |ddkr |dkr|rdSdStj|rt jd|t d |d vrt jd td dS) N)Fznot a valid python identifierrr%)T)Fz8axis name should should not start or end with underscorez7It is discouraged to use axes names that are keywords: ) stacklevel)rFzQIt is discouraged to use 'axis' as an axis name and will raise an error in future)r isidentifierkeyword iskeywordwarningswarnRuntimeWarning FutureWarning)rLrs rr.z.ParsedExpression.check_axis_name_return_reasons%% 99 !W^^tBx3s{{/{xTT &&  TdTT"  x g!  8rc@t|\}}|S)z Valid axes names are python identifiers except keywords, and additionally should not start or end with underscore )rr.)rLis_valid_reasons rcheck_axis_namez ParsedExpression.check_axis_names! -JJ4PP'rN)F)rrrrrboolrrrGrK staticmethodrr.r\rrrrrs EJdiP.P.P.3P.TP.]aP.P.P.P.d4C4TYZ^`cZcTd\,cd\rr)rSrUtypingrrrrreinopsrr r__annotations__r rrrrrbs44444444444444 3 ) ) ) ) ) ) ) )EEEEEEEEEEr