U
    rh                     @  sH   d dl m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 )SpaceToDepthNone)returnc            
   	   C  s   d \} }}}}d}t jjddgdg|d}tj|tj}t|| ||| ||| |g}t	|ddd	d
ddg}t|| ||d  || || g}	t
||g|	gdd d S )N)   r      r	   r   r   xyinputsoutputs	blocksizer               Ztest_spacetodepthr   r   name)onnxhelper	make_nodenprandomZrandom_sampleastypefloat32ZreshapeZ	transposer   )
bchwshaper   noder
   tmpr    r$   L/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/spacetodepth.pyexport   s      $zSpaceToDepth.exportc                  C  s   t jjddgdgdd} tdddd	dd
gddddddgddddddgddddddggggtj}tdddgdddggdd	d
gdddggdddgdddggdddgdddggggtj}t| |g|gdd d S )Nr   r
   r   r   r   r   r	   r                           r   	   r   
   r                        Ztest_spacetodepth_exampler   )r   r   r   r   arrayr   r   r   )r"   r
   r   r$   r$   r%   export_example    s8    	
zSpaceToDepth.export_exampleN)__name__
__module____qualname__staticmethodr&   r9   r$   r$   r$   r%   r      s   r   )

__future__r   Znumpyr   r   Zonnx.backend.test.case.baser   Zonnx.backend.test.case.noder   r   r$   r$   r$   r%   <module>   s
   