o
    wio                     @  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                   @  s(   e Zd ZedddZedddZdS )		TransposereturnNonec                  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outputstest_transpose_defaultr   r   name)
nprandomrandom_sampleastypefloat32onnxhelper	make_node	transposer   )shaper   noder    r   b/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/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 ]#\}}tjjddgdg|d}t ||}t||g|gd| d qd 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   r   r   r   export_all_permutations   s$   z!Transpose.export_all_permutationsN)r   r   )__name__
__module____qualname__staticmethodr    r+   r   r   r   r   r      s
    r   )
__future__r   r$   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r   r   r   r   <module>   s   