
    Si
                    d    S SK Jr  S SKrS SKrS SKJr  S SKJr        SS jr	 " S S\5      r
g)	    )annotationsN)Base)expectc                0    [         R                  " U /UQ76 $ )N)npeinsum)EqnOperandss     a/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/einsum.pyeinsum_reference_implementationr      s     99S$8$$    c                      \ rS rSr\S
S j5       r\S
S j5       r\S
S j5       r\S
S j5       r\S
S j5       r	\S
S j5       r
Srg	)Einsum   c                     Sn [         R                  R                  SS/S/U S9n[        R                  R                  SS5      n[        X45      n[        X/U/SS	9  g )
Nzij->jir   xyinputsoutputsequation      test_einsum_transposer   r   nameonnxhelper	make_noder   randomrandnr   r   )r	   nodeXYs       r   export_einsum_transposeEinsum.export_einsum_transpose   s`    {{$$cUSEC % 
 IIOOAq!+C6tC!3JKr   c                     Sn [         R                  R                  SS/S/U S9n[        R                  R                  SS5      n[        X45      n[        X/U/SS	9  g )
Nzij->ir   r   r   r   r   r   test_einsum_sumr   r   r	   r#   r$   Zs       r   export_einsum_sumEinsum.export_einsum_sum    s`    {{$$cUSEC % 
 IIOOAq!+C6tC!3DEr   c                     Sn [         R                  R                  SS/S/U S9n[        R                  R                  SSS5      n[        X45      n[        X/U/SS	9  g )
Nz...ii ->...ir   r   r   r   r      test_einsum_batch_diagonalr   r   r*   s       r   export_einsum_batch_diagonal#Einsum.export_einsum_batch_diagonal,   sb    {{$$cUSEC % 
 IIOOAq!$+C6tC!3OPr   c                     Sn [         R                  R                  SSS/S/U S9n[        R                  R                  S5      n[        R                  R                  S5      n[        XU45      n[        XU/U/SS	9  g )
Nzi,ir   r   r   zr   r/   test_einsum_inner_prodr   r   r	   r#   r$   r%   r+   s        r   export_einsum_inner_prodEinsum.export_einsum_inner_prod8   sv    {{$$c3Z# % 
 IIOOAIIOOA+CQ8tFQC6NOr   c                    Sn [         R                  R                  SSS/S/U S9n[        R                  R                  SSS	5      n[        R                  R                  SS	S
5      n[        XU45      n[        XU/U/SS9  g )Nzbij, bjk -> bikr   r   r   r4   r   r/      r   r   test_einsum_batch_matmulr   r   r6   s        r   export_einsum_batch_matmul!Einsum.export_einsum_batch_matmulE   s~    {{$$c3Z# % 
 IIOOAq!$IIOOAq!$+CQ8tFQC6PQr   c                     Sn [         R                  R                  SS/S/U S9n[        R                  " S5      n[        X45      n[        X/U/SS9  g )	Nz->r   r   r   r   g      @test_einsum_scalarr   )r   r   r    r   arrayr   r   r*   s       r   export_einsum_scalarEinsum.export_einsum_scalarR   sY    {{$$cUSEC % 
 HHSM+C6tC!3GHr    N)returnNone)__name__
__module____qualname____firstlineno__staticmethodr&   r,   r1   r7   r<   rA   __static_attributes__rC   r   r   r   r      s    	L 	L 	F 	F 	Q 	Q 
P 
P 
R 
R 	I 	Ir   r   )r	   strr
   ztuple[np.ndarray, ...]rD   z
np.ndarray)
__future__r   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r   rC   r   r   <module>rQ      sC    #   , .%	%.%%IIT IIr   