
    
NgW                     n    d dl Zd dlmZ  G d de          Z G d de          Z G d de          ZdS )	    N)OpRunc                       e Zd ZddZdS )Unsqueeze_1Nc                 H   t          |t          j                  rt          |          }n-|g dfv rd }n$t          |t                    rt          |          }t          |t          t          f          r|}|D ]}t          j        ||          }nt          d          |fS )N axisz9axes cannot be None for operator Unsqueeze (Unsqueeze_1).)
isinstancenpndarraytuplelistexpand_dimsRuntimeErrorselfdataaxessqas        [/var/www/html/ai-engine/env/lib/python3.11/site-packages/onnx/reference/ops/op_unsqueeze.py_runzUnsqueeze_1._run   s    dBJ'' 	;;DDb"XDDd## 	;;DdUDM** 	B 0 0^BQ///0 K   u    N__name__
__module____qualname__r   r   r   r   r   r      s(             r   r   c                       e Zd ZdS )Unsqueeze_11N)r   r   r   r   r   r   r    r       s        Dr   r    c                       e Zd ZddZdS )Unsqueeze_13Nc                    |t          |d          rt          |j                  dk    r	 t          j        |t          |                    }n# t          $ rj t          |          dk    r*t          j        |t          |          d                   }n*|}t          |          D ]}t          j        ||          }Y n*w xY wt          j        ||          }nt          d          |fS )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   zUnsqueeze_13._run#   s    tZ(( 5S__q-@-@	<5;;???BB  < < <4yyA~~^DuT{{1~FFF!!)$ < <A!#!;!;!;BB< ^Dt444L   us   #A A1CCr   r   r   r   r   r"   r"   "   s(             r   r"   )numpyr   onnx.reference.op_runr   r   r    r"   r   r   r   <module>r-      s        ' ' ' ' ' '    %   &	 	 	 	 	; 	 	 	    5     r   