
    Sio                    V    S SK Jr  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)    )annotationsN)Base)expectc                  <    \ rS rSr\SS j5       r\SS j5       rSrg)	Transpose   c                    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       d/var/www/html/ai-image-ml/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#    N)returnNone)__name__
__module____qualname____firstlineno__staticmethodr!   r/   __static_attributes__r1   r#   r    r   r      s*    	Y 	Y  r#   r   )
__future__r   r(   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r1   r#   r    <module>r>      s%    #    , .   r#   