o
    toi8                     @  sT   d dl mZ d dlZd dlmZ G dd deZG dd deZG dd	 d	eZdS )
    )annotationsN)OpRunc                   @     e Zd ZdddZdS )Unsqueeze_1Nc                 C  sr   t |tjrt|}n|g dfv rd }n	t |trt|}t |ttfr5|}|D ]	}tj||d}q(|fS td)N axisz9axes cannot be None for operator Unsqueeze (Unsqueeze_1).)
isinstancenpndarraytuplelistexpand_dims	TypeErrorselfdataaxessqar   r   S/home/ubuntu/.local/lib/python3.10/site-packages/onnx/reference/ops/op_unsqueeze.py_run   s   

zUnsqueeze_1._runN__name__
__module____qualname__r   r   r   r   r   r          r   c                   @  s   e Zd ZdS )Unsqueeze_11N)r   r   r   r   r   r   r   r      s    r   c                   @  r   )Unsqueeze_13Nc                 C  s   |d urYt |drOt|jdkrOztj|t|d}W |fS  tyN   t|dkr6tj|t|d d}n|}t|D ]}tj||d}q<Y |fS Y |fS w tj||d}|fS td)N__iter__r   r      z:axes cannot be None for operator Unsqueeze (Unsqueeze_13).)	hasattrlenshaper
   r   r   r   reversedRuntimeErrorr   r   r   r   r   !   s*   	zUnsqueeze_13._runr   r   r   r   r   r   r       r   r   )	
__future__r   numpyr
   onnx.reference.op_runr   r   r   r   r   r   r   r   <module>   s   