
    hp                     B    S SK rS SKrS SKJr  S SKJr   " S S\5      rg)    N)Base)expectc                   <    \ rS rSr\SS j5       r\SS j5       rSrg)SpaceToDepth   Nc            	         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
             b/var/www/fran/franai/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.   r/   )r5   r6   r7   )r   r1   r3   )r8   r:   r<   )r0   r2   r4   )r9   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,    )returnN)__name__
__module____qualname____firstlineno__staticmethodr*   r@   __static_attributes__rB   r,   r)   r   r      s,    H H" !P !Pr,   r   )numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rB   r,   r)   <module>rM      s!   
   , .5P4 5Pr,   