zjBddlTddlmZeddZdS))*)register_decompzpd_op.unsqueezec: |d}t|j}t|}|D]5}|dkr|t |dzz }g|d|d||d}6t ||}|dgS)z%define composite rule of op unsqueezevaluerN)get_defining_opattrslistshapelenreshape)xaxisx_shape axis_listiouts j/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/paddle/decomposition/rules.py unsqueezers2    ! ! ' ' ) )' 2D17mmGT I 22 q55 W! !A1GBQBK11WQRR[1 !W  C ;N) primitivesregisterrrrrrsV %%%%%% "##  $#   r