kh,lddlZddlmZGddejZGddejZy)Nc`eZdZUejj eed<dfd ZdZ xZ S) SpaceToDepthbsc>t||dk(sJ||_y)Nsuper__init__rself block_size __class__s V/var/www/teggl/fontify/venv/lib/python3.12/site-packages/timm/layers/space_to_depth.pyr zSpaceToDepth.__init__s! Qc|j\}}}}|j||||jz|j||jz|j}|jddddddj }|j|||jz|jz||jz||jz}|S)Nrrsizeviewrpermute contiguousr xNCHWs rforwardzSpaceToDepth.forward sVVX 1a FF1adggtwwTWW dgg F IIaAq!Q ' 2 2 4 FF1a$''kDGG+Q$''\1< Hr)r) __name__ __module__ __qualname__torchjitFinalint__annotations__r r! __classcell__rs@rrrs   rrc$eZdZfdZdZxZS) DepthToSpacec0t|||_y)Nrr s rr zDepthToSpace.__init__s rch|j\}}}}|j||j|j||jdzz||}|jddddddj }|j|||jdzz||jz||jz}|S)Nrrrrrrrrs rr!zDepthToSpace.forwardsVVX 1a FF1dggtwwdggl(;Q B IIaAq!Q ' 2 2 4 FF1aDGGqL)1tww;DGG Dr)r"r#r$r r!r*r+s@rr-r-s rr-)r%torch.nnnnModulerr-rrr4s+  299  299 r