
    hW                     Z    S SK rS SKJr   " S S\5      r " S S\5      r " S S\5      rg)	    N)OpRunc                       \ rS rSrSS jrSrg)Unsqueeze_1   Nc                 >   [        U[        R                  5      (       a  [        U5      nO+U/ S4;   a  S nO [        U[        5      (       a  [        U5      n[        U[        [        45      (       a"  UnU H  n[        R
                  " X4S9nM     U4$ [        S5      e)N axisz9axes cannot be None for operator Unsqueeze (Unsqueeze_1).)
isinstancenpndarraytuplelistexpand_dimsRuntimeErrorselfdataaxessqas        Y/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/reference/ops/op_unsqueeze.py_runUnsqueeze_1._run   s    dBJJ'';Db"XDd##;DdUDM**B^^B/  u K     r   N__name__
__module____qualname____firstlineno__r   __static_attributes__r   r   r   r   r      s    r   r   c                       \ rS rSrSrg)Unsqueeze_11   r   N)r   r   r    r!   r"   r   r   r   r$   r$      s    r   r$   c                       \ rS rSrSS jrSrg)Unsqueeze_13"   Nc                    Ubd  [        US5      (       a;  [        UR                  5      S:  a"   [        R                  " U[        U5      S9nU4$ [        R                  " XS9n U4$ [        S5      e! [         aa    [        U5      S:X  a%  [        R                  " U[        U5      S   S9n U4$ Un[        U5       H  n[        R                  " X4S9nM      U4$ f = f)N__iter__r   r	      z:axes cannot be None for operator Unsqueeze (Unsqueeze_13).)	hasattrlenshaper   r   r   	TypeErrorreversedr   r   s        r   r   Unsqueeze_13._run#   s    tZ((S_q-@	<5;?B u ^^D4
 u L  ! <4yA~^^DuT{1~F u "!)$A!#!;B "0 u<s   A3 3:C1(CCr   r   r   r   r   r   r'   r'   "   s    r   r'   )numpyr   onnx.reference.op_runr   r   r$   r'   r   r   r   <module>r4      s1     '% &	; 	5 r   