RЦi,lSSKrSSKJr "SS\R5r"SS\R5rg)Ncx^\rSrSr%Sr\R R\\ S'SS\4U4Sjjjr Sr Sr U=r $) SpaceToDepthzRearrange spatial dimensions into channel dimension. Divides spatial dimensions by block_size and multiplies channels by block_size^2. Used in TResNet as an efficient stem operation. Args: block_size: Spatial reduction factor. bs block_sizec>>[TU]5 US:XdeXlg)Nsuper__init__rselfr __class__s Y/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/timm/layers/space_to_depth.pyr SpaceToDepth.__init__s QcUR5up#pEURX#X@R-URXPR-UR5nURSSSSSS5R 5nURX#UR-UR-X@R-XPR-5nU$)Nrrr sizeviewrpermute contiguousrxNCHWs rforwardSpaceToDepth.forwardsVVX a FF1ggtwwWW dgg F IIaAq!Q ' 2 2 4 FF1$''kDGG+Q''\1< Hrr)r )__name__ __module__ __qualname____firstlineno____doc__torchjitFinalint__annotations__r r"__static_attributes__ __classcell__rs@rrrs8  3 rrc2^\rSrSrSrU4SjrSrSrU=r$) DepthToSpacezRearrange channel dimension into spatial dimensions. Inverse of SpaceToDepth. Divides channels by block_size^2 and multiplies spatial dimensions by block_size. Args: block_size: Spatial expansion factor. c.>[TU]5 Xlg)Nr r s rr DepthToSpace.__init__'s rcXUR5up#pEURX RURX0RS--XE5nURSSSSSS5R 5nURX#URS--X@R-XPR-5nU$)Nrrrr rrrrs rr"DepthToSpace.forward+sVVX a FF1ggtwwggl(;Q B IIaAq!Q ' 2 2 4 FF1DGGqL)1ww;GG Drr$) r%r&r'r(r)r r"r/r0r1s@rr3r3srr3)r*torch.nnnnModulerr3rrr=s+ 2990299r