ó
    ±“,jº  ã                   ó   • S  r S rg)c           	      óî   ^^^^^• SmUU4S jmSR                  U4S j[        U 5       5       5      nSmUU4S jmSR                  U4S j[        U [        T5      -
  S-   5       5       5      nUS-   U-   $ )Né   c                 óN   >• U T;   a  g[        [        S5      T-   5      nTS-  mU$ ©NÚar   ©ÚchrÚord)ÚidxÚresultÚ	_next_ordÚaxes_ins     €€ÚµC:\Users\johnn\AppData\Local\Packages\PythonSoftwareFoundation.Python.3.13_qbz5n2kfra8p0\LocalCache\local-packages\Python313\site-packages\einx/_src/adapter/classical_from_einsum.pyÚget_input_name_atÚ._einsum_diag_string.<locals>.get_input_name_at   s.   ø€ à�'‹>Øäœ˜S› IÑ-Ó.ˆFØ˜‰NˆIØˆMó    Ú c              3   ó4   >#   • U  H  nT" U5      v •  M     g 7f©N© )Ú.0Úir   s     €r   Ú	<genexpr>Ú&_einsum_diag_string.<locals>.<genexpr>   s   øé € ÐIº[¸Ñ0°×3Ð3º[ùó   ƒc                 óN   >• U T:X  a  g[        [        S5      T-   5      nTS-  mU$ r   r   )r
   r   r   Úaxis_outs     €€r   Úget_output_name_atÚ/_einsum_diag_string.<locals>.get_output_name_at   s.   ø€ à�(‹?Øäœ˜S› IÑ-Ó.ˆFØ˜‰NˆIØˆMr   c              3   ó4   >#   • U  H  nT" U5      v •  M     g 7fr   r   )r   r   r   s     €r   r   r      s   øé € Ð^Ò?]¸!Ñ 2°1× 5Ð 5Ò?]ùr   z->)ÚjoinÚrangeÚlen)Úndimr   r   Úeinsum_str_inputÚeinsum_str_outputr   r   r   s    ``  @@@r   Ú_einsum_diag_stringr&      so   ü€ Ø€Iöð —w‘wÔI¼UÀ4¼[ÓIÓIÐà€Iöð Ÿ™Ô^¼uÀTÌCÐPWËLÑEXÐ[\ÑE\Ô?]Ó^Ó^Ðà˜dÑ"Ð%6Ñ6Ð6r   c                 ó   ^ • U 4S jnU$ )Nc                 ó@   >• [        U R                  X5      nT" X05      $ r   )r&   r#   )Úxr   r   Ú
einsum_strÚeinsums       €r   ÚdiagonalÚdiagonal.<locals>.diagonal    s   ø€ Ü(¨¯©°ÓCˆ
Ù�jÓ$Ð$r   r   )r+   r,   s   ` r   r,   r,      s   ø€ õ%ð €Or   N)r&   r,   r   r   r   Ú<module>r.      s   ðò7ó<r   