
    wiT                     R    d dl Z d dlmc mZ d dl mZ  G d dej                  ZdS )    N)nnc                   (     e Zd ZdZ fdZd Z xZS )ClsHeadzd
    Class orientation
    Args:
        params(dict): super parameters for build Class network
    c                     t          t          |                                            t          j        d          | _        t          j        ||d          | _        d S )N   T)bias)superr   __init__r   AdaptiveAvgPool2dpoolLinearfc)selfin_channels	class_dimkwargs	__class__s       x/var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/rapidocr/networks/heads/cls_head.pyr
   zClsHead.__init__   sM    gt%%'''(++	)K>>>    c                     |                      |          }t          j        ||j        d         |j        d         g          }|                     |          }t          j        |d          }|S )Nr   r   )shape)dim)r   torchreshaper   r   Fsoftmax)r   xs     r   forwardzClsHead.forward   s^    IIaLLM!AGAJ
#;<<<GGAJJIaQr   )__name__
__module____qualname____doc__r
   r   __classcell__)r   s   @r   r   r      sQ         ? ? ? ? ?
      r   r   )r   torch.nn.functionalr   
functionalr   Moduler    r   r   <module>r(      sq                       bi     r   