U
    qh`                     @  s4   d dl mZ d dlZd dlmZ G dd deZdS )    )annotationsN)OpRunc                   @  s   e Zd ZdddZdS )	TransposeNc                 C  sj   |d kst |dkrd n|}|d kr0t|fS t |t |jkrZtd|d|jdtj||dfS )Nr   zInconsistent permutation z with shape .)Zaxes)lennpZ	transposeshapeRuntimeError)selfdatapermZperm_ r   C/tmp/pip-unpacked-wheel-xnis5xre/onnx/reference/ops/op_transpose.py_run   s    zTranspose._run)N)__name__
__module____qualname__r   r   r   r   r   r      s   r   )
__future__r   Znumpyr   Zonnx.reference.op_runr   r   r   r   r   r   <module>   s   