U
    rho                     @  sP   d dl mZ 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 deZ
dS )    )annotationsN)Base)expectc                   @  s0   e Zd ZeddddZeddddZdS )	TransposeNone)returnc                  C  sP   d} t j| t j}tjjddgdgd}t |}t	||g|gdd d S )N         r   data
transposed)inputsoutputsZtest_transpose_defaultr   r   name)
nprandomrandom_sampleastypefloat32onnxhelper	make_node	transposer   )shaper   noder    r   I/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/transpose.pyexport_default   s      
zTranspose.export_defaultc                  C  s   d} t j| t j}ttt t	| }t
|D ]F\}}tjjddgdg|d}t ||}t||g|gd| d q8d S )Nr   r   r   r   )r   r   permZ test_transpose_all_permutations_r   )r   r   r   r   r   list	itertoolspermutationsZarangelen	enumerater   r   r   r   r   )r   r   r#   iZpermutationr   r   r   r   r   export_all_permutations   s"    z!Transpose.export_all_permutationsN)__name__
__module____qualname__staticmethodr   r'   r   r   r   r   r      s   r   )
__future__r   r"   Znumpyr   r   Zonnx.backend.test.case.baser   Zonnx.backend.test.case.noder   r   r   r   r   r   <module>   s   