o پi @s0ddlZ   d dejdededefd d ZdS) NtokenFx pool_typenum_prefix_tokensreduce_include_prefixcCs|s|S|dkr|dddf}|S|r|n |dd|df}|dkr,|jdd}|S|dkr@d|jdd|jdd}|S|dkrL|jdd}|S|rUJd ||S) Nrravgr)dimavgmaxg?maxzUnknown pool type )meanamax)rrrrrF/home/ubuntu/.local/lib/python3.10/site-packages/timm/layers/pool1d.pyglobal_pool_nlcs"   r)rrF)torchTensorstrintboolrrrrrs