o ÓÙ¾i,ã@s<ddlZddlmZGdd„dejƒZGdd„dejƒZdS)éNcs8eZdZUejjeed<d‡fdd„ Zdd„Z ‡Z S)Ú SpaceToDepthÚbsécs tƒ ¡|dks J‚||_dS)Nr©ÚsuperÚ__init__r©ÚselfÚ block_size©Ú __class__©úN/home/ubuntu/.local/lib/python3.10/site-packages/timm/layers/space_to_depth.pyrs   zSpaceToDepth.__init__cCsx| ¡\}}}}| ||||j|j||j|j¡}| dddddd¡ ¡}| |||j|j||j||j¡}|S)Nréééér©ÚsizeÚviewrÚpermuteÚ contiguous©r ÚxÚNÚCÚHÚWr r rÚforward s $(zSpaceToDepth.forward)r) Ú__name__Ú __module__Ú __qualname__ÚtorchÚjitÚFinalÚintÚ__annotations__rrÚ __classcell__r r r rrs rcs$eZdZ‡fdd„Zdd„Z‡ZS)Ú DepthToSpacecstƒ ¡||_dS)Nrrr r rrs  zDepthToSpace.__init__cCst| ¡\}}}}| ||j|j||jd||¡}| dddddd¡ ¡}| |||jd||j||j¡}|S)Nrrrrrrrrr r rrs "&zDepthToSpace.forward)rr r!rrr'r r r rr(s r()r"Útorch.nnÚnnÚModulerr(r r r rÚs