
    Si                    N    S SK Jr  S SKrS SKrS SKJr  S SKJr   " S S\5      r	g)    )annotationsN)Base)expectc                  <    \ rS rSr\SS j5       r\SS j5       rSrg)SpaceToDepth   c            	        S=u  pp#nSn[         R                  R                  SS/S/US9n[        R                  R                  U5      R                  [        R                  5      n[        R                  " XpXU-  XSU-  U/5      n[        R                  " U/ SQ5      n[        R                  " XXS-  -  X%-  X5-  /5      n	[        Xg/U	/SS	9  g )
N)   r
      r   r
   r   xyinputsoutputs	blocksize)r            r
      test_spacetodepthr   r   name)onnxhelper	make_nodenprandomrandom_sampleastypefloat32reshape	transposer   )
bchwshaper   noder   tmpr   s
             g/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/spacetodepth.pyexportSpaceToDepth.export   s    ))
aU	{{$$5E	 % 
 II##E*11"**=jj19niiK
 ll3 23JJs\ 2ANANSTtC!3FG    c                    [         R                  R                  SS/S/SS9n [        R                  " / SQ/ SQ/ SQ/ S	Q///5      R                  [        R                  5      n[        R                  " / S
Q/ SQ// SQ/ SQ// SQ/ SQ// SQ/ SQ///5      R                  [        R                  5      n[        X/U/SS9  g )Nr   r   r   r
   r   )r   r   r      r
      )      r            )r   	   r   
   r      )                  )r   r   r
   )r   r   r   )r   r/   r0   )r6   r7   r8   )r1   r   r4   )r9   r;   r=   )r2   r3   r5   )r:   r<   r>   test_spacetodepth_exampler   )r   r   r   r   arrayr   r    r   )r(   r   r   s      r*   export_exampleSpaceToDepth.export_example    s    {{$$5E	 % 
 HH +0,0		
 &
 	
 HH 	*,!<0!<0		
 &
 	
 	tC!3NOr-    N)returnNone)__name__
__module____qualname____firstlineno__staticmethodr+   rA   __static_attributes__rC   r-   r*   r   r      s,    H H" !P !Pr-   r   )

__future__r   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rC   r-   r*   <module>rP      s$    #   , .5P4 5Pr-   