a Sic)@sdZddlZddlmZddlZgdZdZddZdd Z d d Z d d Z ddZ ddZ ddZddZddZddZddZddZdS)zC A functionally equivalent parser of the numpy.einsum input parser N) OrderedDict) is_valid_einsum_charhas_valid_einsum_chars_only get_symbolgen_unused_symbolsconvert_to_valid_einsum_charsalpha_canonicalizefind_output_strfind_output_shapepossibly_convert_to_numpyparse_einsum_input4abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZcCs|tvp|dvS)uCheck if the character ``x`` is valid for numpy einsum. Examples -------- >>> is_valid_einsum_char("a") True >>> is_valid_einsum_char("Ǵ") False z,->.)_einsum_symbols_basexrM/var/www/html/django/DPS/env/lib/python3.9/site-packages/opt_einsum/parser.pyrs rcCsttt|S)uCheck if ``einsum_str`` contains only valid characters for numpy einsum. Examples -------- >>> has_valid_einsum_chars_only("abAZ") True >>> has_valid_einsum_chars_only("Över") False )allmapr) einsum_strrrrr#s rcCs|dkrt|St|dS)uGet the symbol corresponding to int ``i`` - runs through the usual 52 letters before resorting to unicode characters, starting at ``chr(192)``. Examples -------- >>> get_symbol(2) 'c' >>> get_symbol(200) 'Ŕ' >>> get_symbol(20000) '京' 4)rchr)irrrr1srccs>d}}||kr:t|}|d7}||vr*q|V|d7}qdS)zGenerate ``n`` symbols that are not already in ``used``. Examples -------- >>> list(oe.parser.gen_unused_symbols("abd", 2)) ['c', 'e'] rNr)usednrcntsrrrrEsrcs>tt|td}ddt|Ddfdd|DS)uConvert the str ``einsum_str`` to contain only the alphabetic characters valid for numpy einsum. If there are too many symbols, let the backend throw an error. Examples -------- >>> oe.parser.convert_to_valid_einsum_chars("Ĥěļļö") 'cbdda' z,->cSsi|]\}}|t|qSrr).0rrrrr bz1convert_to_valid_einsum_chars..c3s|]}||VqdSNgetr rreplacerrr cr"z0convert_to_valid_einsum_chars..)sortedset enumeratejoin)rsymbolsrr(rrWs rcsJt|D]&}|dvrq |vr tt|<q dfdd|DS)uAlpha convert an equation in an order-independent canonical way. Examples -------- >>> oe.parser.alpha_canonicalize("dcba") 'abcd' >>> oe.parser.alpha_canonicalize("Ĥěļļö") 'abccd' z.,->r#c3s|]}||VqdSr$r%r'renamerrr*wr"z%alpha_canonicalize..)rrlenr.)equationnamerr0rrfs rcs,|dddfddttDS)aU Find the output string for the inputs ``subscripts`` under canonical einstein summation rules. That is, repeated indices are summed over by default. Examples -------- >>> oe.parser.find_output_str("ab,bc") 'ac' >>> oe.parser.find_output_str("a,b") 'ab' >>> oe.parser.find_output_str("a,a,b,b") '' ,r#c3s |]}|dkr|VqdS)rN)count)r rtmp_subscriptsrrr*r"z"find_output_str..)replacer.r+r,) subscriptsrr7rr zs r cstfdd|DS)aOFind the output shape for given inputs, shapes and output string, taking into account broadcasting. Examples -------- >>> oe.parser.find_output_shape(["ab", "bc"], [(2, 3), (3, 4)], "ac") (2, 4) # Broadcasting is accounted for >>> oe.parser.find_output_shape(["a", "a"], [(4, ), (1, )], "a") (4,) c3s4|],tddtfddDDVqdS)css"|]\}}|dkr||VqdS)rNr)r shapelocrrrr*r"z.find_output_shape...csg|]}|qSr)findr'crr r"z/find_output_shape...N)maxzip)r inputsshapesr>rr*sz$find_output_shape..)tuple)rDrEoutputrrCrr s r cCst|dst|S|SdS)aTConvert things without a 'shape' to ndarrays, but leave everything else. Examples -------- >>> oe.parser.possibly_convert_to_numpy(5) array(5) >>> oe.parser.possibly_convert_to_numpy([5, 3]) array([5, 3]) >>> oe.parser.possibly_convert_to_numpy(np.array([5, 3])) array([5, 3]) # Any class with a shape is passed through >>> class Shape: ... def __init__(self, shape): ... self.shape = shape ... >>> myshape = Shape((5, 5)) >>> oe.parser.possibly_convert_to_numpy(myshape) <__main__.Shape object at 0x10f850710> r;N)hasattrnp asanyarrayrrrrr s  r cCs0d}|D]"}|tur|d7}q|||7}q|S)aConvert user custom subscripts list to subscript string according to `symbol_map`. Examples -------- >>> oe.parser.convert_subscripts(['abc', 'def'], {'abc':'a', 'def':'b'}) 'ab' >>> oe.parser.convert_subscripts([Ellipsis, object], {object:'a'}) '...a' r#...)Ellipsis)Zold_sub symbol_mapZnew_subrrrrconvert_subscriptss  rNcst|}g}g}tt|dD]$}||d||dq t|rV|dnd}dd|D}z4ttj|}| t ddt t |DWnt yt d Yn0d fd d |D}|dur|d 7}|t|7}||fS)z:Convert 'interleaved' input to standard einsum input. rNcSsg|] }t|qSrr r'rrrr@r"z-convert_interleaved_input..cSsi|]\}}|t|qSrr)r idxsymbolrrrr!r"z-convert_interleaved_input..ziFor this input type lists must contain either Ellipsis or hashable and comparable object (e.g. int, str).r5c3s|]}t|VqdSr$)rN)r subrMrrr*r"z,convert_interleaved_input..->)listranger2appendpopr, itertoolschain from_iterablediscardrLr-r+ TypeErrorr.rN)operands tmp_operands operand_listsubscript_listp output_listZ symbol_setr:rrUrconvert_interleaved_inputs&  rfcCst|dkrtdt|dtrJ|ddd}dd|ddD}n t|\}}d |vsfd |vr|d dkp|d dk}|s|d dkrtd d |vrd|d dddd d}dt|t dd|D}d}d |vr | d \}}| d}d} n| d}d} t |D]\} } d | vr"| d dksT| ddkr\td|| j dkrrd} n t t|| j dt| d} | |kr| }| dkrtdn6| dkr| dd|| <n| d|| d|| <q"d|}|dkrd} n|| d} | r2|d |d| 7}n2t |}dtt|t| }|d | |7}d |vr~| d \}}n|t |}}|D]}||vrtd|qt| dt|krtd|||fS)af A reproduction of einsum c side einsum parsing in python. Returns ------- input_strings : str Parsed input strings output_string : str Parsed output string operands : list of array_like The operands to use in the numpy contraction Examples -------- The operand list is simplified to reduce printing: >>> a = np.random.rand(4, 4) >>> b = np.random.rand(4, 4, 4) >>> parse_einsum_input(('...a,...a->...', a, b)) ('za,xza', 'xz', [a, b]) >>> parse_einsum_input((a, [Ellipsis, 0], b, [Ellipsis, 0])) ('za,xza', 'xz', [a, b]) rzNo input operands r#cSsg|] }t|qSrrQr'rrrr@r"z&parse_einsum_input..rN->rVz%Subscripts can only contain one '->'..r5css|]}t|jVqdSr$)r2r;r'rrrr*#r"z%parse_einsum_input..TFrKzInvalid Ellipses.rzEllipses lengths do not match.z1Output character '{}' did not appear in the inputzDNumber of einsum subscripts must be equal to the number of operands.)r2 ValueError isinstancestrr9rfr6r.rrAsplitr-r;r r+r,format)r`r:invalidr ellipse_indslongest input_tmp output_subsplit_subscriptsout_subnumrT ellipse_count out_ellipseoutput_subscript normal_indsinput_subscriptscharrrrr sh                 r )__doc__r[ collectionsrnumpyrI__all__rrrrrrrr r r rNrfr rrrrs" %