^jJ @ddlZ ddejdededefdZy)Nx pool_typenum_prefix_tokensreduce_include_prefixc&|s|S|dk(r |dddf}|S|r|n |dd|df}|dk(r|jd}|S|dk(r*d|jd|jdzz}|S|d k(r|jd}|S|r Jd ||S) aApply global pooling to tensor in NLC format. Args: x: Input tensor in (batch, length, channels) format. pool_type: Pooling type - 'token', 'avg', 'max', 'avgmax', or empty string. num_prefix_tokens: Number of prefix tokens (e.g., class token) to exclude from pooling. reduce_include_prefix: Whether to include prefix tokens in reduction. Returns: Pooled tensor. tokenNravg)dimavgmaxg?maxzUnknown pool type )meanamax)rrrrs ]/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/timm/layers/pool1d.pyglobal_pool_nlcrs" G adG H'AAa1B1C.C,D  1 A H( "qvv!v}qvv!v}45A H % 1 A H! B$6yk"B B= H)rr F)torchTensorstrintboolrrrrs@ !!"&+ ! <<! ! !  $ ! r