o toi@s4ddlmZddlZddlmZGdddeZdS)) annotationsN)OpRunc@seZdZdddZdS) DepthToSpaceNc Cst|jdkrtd|jd|j\}}}}|dkr5||||||||f}||} t| gd} n||||||||f}||} t| gd} ||||||||f} t| | } | fS)NzUnexpected shape .DCR)rr)rr rr r r)lenshape RuntimeErrorreshapenp transpose) selfdata blocksizemodebchwtmpshapereshaped transposed finalshapeyrX/home/ubuntu/.local/lib/python3.10/site-packages/onnx/reference/ops/op_depth_to_space.py_run s:      zDepthToSpace._run)NN)__name__ __module__ __qualname__r!rrrr r sr) __future__rnumpyronnx.reference.op_runrrrrrr s