RЦiXSSKJr SSKJr SSKr"SS\\5r\\\4rS\4SjrS\4Sjr S \RS\4S jr S \RS\4S jr g) )Enum)UnionNc$\rSrSrSrSrSrSrSrg)FormatNCHWNHWCNCLNLCN) __name__ __module__ __qualname____firstlineno__rr r r __static_attributes__r Q/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/timm/layers/format.pyrrs D D C Crrfmtc[U5nU[RLaSnU$U[RLaSnU$U[RLaSnU$SnU$)zReturn spatial dimension indices for a given tensor format. Args: fmt: Tensor format (NCHW, NHWC, NCL, or NLC). Returns: Tuple of spatial dimension indices. )))rr)r)rr r r rdims rget_spatial_dimrse +C fjj J   J   J Jrc|[U5nU[RLaSnU$U[RLaSnU$SnU$)zReturn channel dimension index for a given tensor format. Args: fmt: Tensor format (NCHW, NHWC, NCL, or NLC). Returns: Channel dimension index. rrr)rr r rs rget_channel_dimr&sK +C fkk J   J JrxcU[R:XaURSSSS5nU$U[R:Xa#UR S5R SS5nU$U[R :XaUR S5nU$)zConvert tensor from NCHW format to specified format. Args: x: Input tensor in NCHW format. fmt: Target format. Returns: Tensor in target format. rrrr)rr permuter flatten transposer rrs rnchw_tor$9sz fkk IIaAq ! H   IIaL " "1a ( H   IIaL HrcU[R:XaURSSSS5nU$U[R:XaUR SS5nU$U[R :Xa"UR SS5R SS5nU$)zConvert tensor from NHWC format to specified format. Args: x: Input tensor in NHWC format. fmt: Target format. Returns: Tensor in target format. rrrr)rrr r r!r r"r#s rnhwc_tor&Ls~ fkk IIaAq ! H   IIaO H   IIaO % %a + Hr) enumrtypingrtorchstrrFormatTrrTensorr$r&r rrr-sr S$ V *& u|| & & u|| & r