o
    ü* iè  ã                   @   s(   d dl T d dlmZ edƒdd„ ƒZdS )é   )Ú*)Úregister_decompzpd_op.unsqueezec                 C   sx   	 |  ¡  ¡ d }t| jƒ}t|ƒ}|D ]}|dk r"|t|ƒd 7 }g |d|… ¢d‘||d… ¢}qt| |ƒ}|dgS )z%define composite rule of op unsqueezeÚvalueé    r   N)Zget_defining_opÚattrsÚlistÚshapeÚlenZreshape)ÚxZaxisZx_shapeZ	axis_listÚiÚout© r   úf/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/paddle/decomposition/rules.pyÚ	unsqueeze   s   
"
r   N)Z
primitivesÚregisterr   r   r   r   r   r   Ú<module>   s   