
    Si8                    f    S SK Jr  S SKrS SKJr   " S S\5      r " S S\5      r " S S	\5      rg)
    )annotationsN)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	TypeErrorselfdataaxessqas        ^/var/www/html/ai-image-ml/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 WXX    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   r   reversedRuntimeErrorr   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(   )	
__future__r   numpyr   onnx.reference.op_runr   r   r%   r(   r	   r   r   <module>r6      s4    #  '% "	; 	5 r   