o tois@s4ddlmZddlZddlmZGdddeZdS)) annotationsN)OpRunc@seZdZdddZdS) SpaceToDepthNc Cst|jdkrtd|jd|j\}}}}||||||||f}t||}t|gd} ||||||||f} t| | |j} | fS)NzUnexpected shape .)rr)lenshape RuntimeErrornpreshape transposeastypedtype) selfdata blocksizebCHWtmpshapereshaped transposed finalshapeyrX/home/ubuntu/.local/lib/python3.10/site-packages/onnx/reference/ops/op_space_to_depth.py_run s&  zSpaceToDepth._run)N)__name__ __module__ __qualname__r!rrrr r sr) __future__rnumpyronnx.reference.op_runrrrrrr s