khUddlmZddlmZddlZGddeeZeeefZdefdZdefdZ d ejdefd Z d ejdefd Z y) )Enum)UnionNceZdZdZdZdZdZy)FormatNCHWNHWCNCLNLCN)__name__ __module__ __qualname__rrr r N/var/www/teggl/fontify/venv/lib/python3.12/site-packages/timm/layers/format.pyrrs D D C Crrfmtct|}|tjurd}|S|tjurd}|S|tjurd}|Sd}|S)N)))rr)r)rr r rrdims rget_spatial_dimrsc +C fjj J   J   J Jrcxt|}|tjurd}|S|tjurd}|Sd}|S)Nrrr)rrr rs rget_channel_dimrsI +C fkk J   J Jrxc |tjk(r|jdddd}|S|tjk(r#|j dj dd}|S|tj k(r|j d}|S)Nrrrr)rrpermuter flatten transposer rrs rnchw_tor!)sx fkk IIaAq ! H   IIaL " "1a ( H   IIaL Hrc|tjk(r|jdddd}|S|tjk(r|j dd}|S|tj k(r"|j ddj dd}|S)Nrrrr)rrrr rr rr s rnhwc_tor#3s| fkk IIaAq ! H   IIaO H   IIaO % %a + Hr) enumrtypingrtorchstrrFormatTrrTensorr!r#rrrr*sr S$ V    u|| &  u|| & r