Ng,fddlZddlmZGddejZGddejZdS)NcReZdZUejjeed<dfd ZdZ xZ S) SpaceToDepthbscft|dksJ||_dS)Nrsuper__init__rself block_size __class__s V/var/www/html/ai-engine/env/lib/python3.11/site-packages/timm/layers/space_to_depth.pyr zSpaceToDepth.__init__s1 Qch|\}}}}|||||jz|j||jz|j}|dddddd}||||jz|jz||jz||jz}|S)Nrrsizeviewrpermute contiguousr xNCHWs rforwardzSpaceToDepth.forward sVVXX 1a FF1adgtwTW dg F F IIaAq!Q ' ' 2 2 4 4 FF1a$'kDG+Q$'\1< H Hr)r) __name__ __module__ __qualname__torchjitFinalint__annotations__r r! __classcell__rs@rrrsc  rrc$eZdZfdZdZxZS) DepthToSpacecVt||_dS)Nrr s rr zDepthToSpace.__init__s$ rcT|\}}}}|||j|j||jdzz||}|dddddd}||||jdzz||jz||jz}|S)Nrrrrrrrrs rr!zDepthToSpace.forwardsVVXX 1a FF1dgtwdgl(;Q B B IIaAq!Q ' ' 2 2 4 4 FF1aDGqL)1tw;DG D Dr)r"r#r$r r!r*r+s@rr-r-sGrr-)r%torch.nnnnModulerr-rrr4s      29        29     r