
    NgL                     R    d dl Z d dlZd dlZd dlmZ d dlmZ  G d de          ZdS )    N)Base)expectc                   >    e Zd Zedd            Zedd            ZdS )	TransposereturnNc                     d} t           j                            |                               t           j                  }t
          j                            ddgdg          }t          j        |          }t          ||g|gd           d S )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       a/var/www/html/ai-engine/env/lib/python3.11/site-packages/onnx/backend/test/case/node/transpose.pyexport_defaultzTranspose.export_default   s    y&&u--44RZ@@{$$<. % 
 
 \$''
tTFZL?WXXXXXX    c            	         d} t           j                            |                               t           j                  }t          t          j        t          j        t          |                                         }t          |          D ]W\  }}t          j                            ddgdg|          }t          j        ||          }t          ||g|gd|            Xd S )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_permutationsz!Transpose.export_all_permutations   s    y&&u--44RZ@@I229SZZ3H3HIIJJ'55 	 	NA{;((x% 	 )  D dK88Jv#;;;	    	 	r!   )r   N)__name__
__module____qualname__staticmethodr    r-    r!   r   r   r      sV        	Y 	Y 	Y \	Y    \  r!   r   )	r&   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r2   r!   r   <module>r6      s   
          , , , , , , . . . . . .                   r!   