
    hL                     J    S SK r S SKrS SKrS SKJr  S SKJr   " S S\5      rg)    N)Base)expectc                   <    \ rS rSr\SS j5       r\SS j5       rSrg)	Transpose   Nc                     Sn [         R                  R                  U 5      R                  [         R                  5      n[
        R                  R                  SS/S/S9n[         R                  " U5      n[        X!/U/SS9  g )N         r   data
transposed)inputsoutputstest_transpose_defaultr   r   name)
nprandomrandom_sampleastypefloat32onnxhelper	make_node	transposer   )shaper   noder   s       _/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/transpose.pyexport_defaultTranspose.export_default   so    yy&&u-44RZZ@{{$$<. % 
 \\$'
tFZL?WX    c            	         Sn [         R                  R                  U 5      R                  [         R                  5      n[        [        R                  " [         R                  " [        U 5      5      5      5      n[        U5       HN  u  p4[        R                  R                  SS/S/US9n[         R                  " X5      n[        UU/U/SU 3S9  MP     g )Nr	   r   r   r   )r   r   perm test_transpose_all_permutations_r   )r   r   r   r   r   list	itertoolspermutationsarangelen	enumerater   r   r   r   r   )r   r   r(   ipermutationr   r   s          r   export_all_permutations!Transpose.export_all_permutations   s    yy&&u-44RZZ@I22299SZ3HIJ'5NA;;((x% 	 ) D d8Jv#7s;	 6r"    )returnN)__name__
__module____qualname____firstlineno__staticmethodr    r.   __static_attributes__r0   r"   r   r   r      s*    	Y 	Y  r"   r   )	r'   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r0   r"   r   <module>r;      s"   
    , .   r"   