
    wi8                    z    d dl mZ 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 )
    )annotationsN)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	TypeErrorselfdataaxessqas        o/root/.openclaw/workspace/chatterbox_venv_py311/lib/python3.11/site-packages/onnx/reference/ops/op_unsqueeze.py_runzUnsqueeze_1._run   s    dBJ'' 	;;DDb"XDDd## 	;;DdUDM** 	YB 0 0^BQ///0 WXXX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   r   reversedRuntimeErrorr   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#   )	
__future__r   numpyr   onnx.reference.op_runr   r   r!   r#   r   r   r   <module>r/      s    # " " " " "     ' ' ' ' ' '    %   "	 	 	 	 	; 	 	 	    5     r   